Searched refs:padded_val (Results 1 – 6 of 6) sorted by relevance
/dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
H A D | masking.py | 44 padded_val, = padded_vals 45 return prim.bind(padded_val, **params)
|
/dports/math/py-jax/jax-0.2.9/jax/ |
H A D | api.py | 1638 def logical_shape(poly_shape, padded_val): argument 1640 return ShapeDtypeStruct(shape, core.get_aval(padded_val).dtype)
|
/dports/math/py-jax/jax-0.2.9/jax/_src/lax/ |
H A D | control_flow.py | 1918 for padded_val, logical_shape in zip(padded_vals, logical_shapes): 1919 result = _memcpy(dimension, logical_shape[dimension], padded_val,
|
H A D | lax.py | 4955 (padded_val,), (logical_shape,) = padded_vals, logical_shapes 4956 padded_shape = masking.padded_shape_as_value(padded_val.shape) 4960 masked_val = select(mask, padded_val, identity(padded_shape, padded_val.dtype))
|
/dports/lang/zig-devel/zig-0.9.0/src/stage1/ |
H A D | codegen.cpp | 7905 LLVMValueRef padded_val[] = { in gen_const_val() local 7909 … fields[type_struct_field->gen_index] = LLVMConstStruct(padded_val, 2, true); in gen_const_val()
|
/dports/lang/zig/zig-0.9.0/src/stage1/ |
H A D | codegen.cpp | 7905 LLVMValueRef padded_val[] = { in gen_const_val() local 7909 … fields[type_struct_field->gen_index] = LLVMConstStruct(padded_val, 2, true); in gen_const_val()
|