Home
last modified time | relevance | path

Searched refs:tensor_list_element_shape (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dlist_ops.py224 element_shape=gen_list_ops.tensor_list_element_shape(
242 element_shape=gen_list_ops.tensor_list_element_shape(
274 element_shape=gen_list_ops.tensor_list_element_shape(
291 gen_list_ops.tensor_list_element_shape(op.inputs[0],
327 element_shape = gen_list_ops.tensor_list_element_shape(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dtensor_list_ops_test.py40 e32 = list_ops.tensor_list_element_shape(l, shape_type=dtypes.int32)
41 e64 = list_ops.tensor_list_element_shape(l, shape_type=dtypes.int64)
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dcontrol_flow_ops_test.py1192 list_ops.tensor_list_element_shape(handle, dtypes.int32),
1193 list_ops.tensor_list_element_shape(handle, dtypes.int64))
1206 list_ops.tensor_list_element_shape(handle, dtypes.int32),
1207 list_ops.tensor_list_element_shape(handle, dtypes.int64))
H A Dpfor.py3950 shape = list_ops.tensor_list_element_shape(handle, shape_type)
3964 element_shape = list_ops.tensor_list_element_shape(handle, dtypes.int32)
4733 element_shape = list_ops.tensor_list_element_shape(
4796 element_shape = list_ops.tensor_list_element_shape(inp, dtypes.int32)
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
H A Dlist_ops_test.py910 element_shape = list_ops.tensor_list_element_shape(
1220 shape = list_ops.tensor_list_element_shape(l, shape_type=dtypes.int32)
1778 l_element_shape = list_ops.tensor_list_element_shape(l, dtypes.int32)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_compiler_test.cc1694 xla::Shape tensor_list_element_shape; in TEST_F() local
1696 &tensor_list_element_shape)); in TEST_F()
1699 std::vector<xla::Shape> shapes{tensor_list_element_shape, index_shape}; in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dmark_for_compilation_pass_test.cc1357 Output tensor_list_element_shape = ops::TensorListElementShape( in TEST() local
1362 tensor_list_element_shape.node()); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
H A Dwhile_v2_test.py1356 shape = gen_list_ops.tensor_list_element_shape(