Searched refs:tensor_list_element_shape (Results 1 – 8 of 8) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
H A D | list_ops.py | 224 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 D | tensor_list_ops_test.py | 40 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 D | control_flow_ops_test.py | 1192 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 D | pfor.py | 3950 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 D | list_ops_test.py | 910 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 D | xla_compiler_test.cc | 1694 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 D | mark_for_compilation_pass_test.cc | 1357 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 D | while_v2_test.py | 1356 shape = gen_list_ops.tensor_list_element_shape(
|