Home
last modified time | relevance | path

Searched refs:padded_val (Results 1 – 6 of 6) sorted by relevance

/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dmasking.py44 padded_val, = padded_vals
45 return prim.bind(padded_val, **params)
/dports/math/py-jax/jax-0.2.9/jax/
H A Dapi.py1638 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 Dcontrol_flow.py1918 for padded_val, logical_shape in zip(padded_vals, logical_shapes):
1919 result = _memcpy(dimension, logical_shape[dimension], padded_val,
H A Dlax.py4955 (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 Dcodegen.cpp7905 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 Dcodegen.cpp7905 LLVMValueRef padded_val[] = { in gen_const_val() local
7909 … fields[type_struct_field->gen_index] = LLVMConstStruct(padded_val, 2, true); in gen_const_val()