| /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
| D | fuse_binary_into_preceding_affine_test.cc | 49 Shape* array_shape = array.mutable_shape(); in CreateArray() local
|
| D | remove_successive_transpose_test.cc | 37 toco::Shape* array_shape = array.mutable_shape(); in CreateArray() local
|
| D | fuse_binary_into_following_affine_test.cc | 49 Shape* array_shape = array.mutable_shape(); in CreateArray() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_concatenation.cc | 91 const Shape array_shape = input_array->shape(); in ConcatenateTensorBuffers() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_test.cc | 87 Shape array_shape = in TEST_F() local
|
| D | shape_util_test.cc | 439 Shape array_shape = ShapeUtil::MakeShape(F32, {42, 42, 123}); in TEST() local
477 Shape array_shape = ShapeUtil::MakeShape(F32, {42, 42, 123}); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shaped_buffer_test.cc | 155 Shape array_shape = ShapeUtil::MakeShape(F32, {1}); in TEST() local
|
| D | pattern_matcher_test.cc | 87 auto array_shape = ShapeUtil::MakeShape(F32, {2, 3, 4}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_execute_test.cc | 282 const Shape array_shape = ShapeUtil::MakeShape(F32, {2, 2}); in XLA_TEST_F() local
328 const Shape array_shape = ShapeUtil::MakeShape(F32, {2, 2}); in XLA_TEST_F() local
371 const Shape array_shape = ShapeUtil::MakeShape(F32, {2, 2}); in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | driver.cc | 439 ArrayShape array_shape = shape.elements[tuple_idx]; in Display() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | dump_graphviz.cc | 282 auto& array_shape = array.shape(); in GetArrayLabel() local
|
| D | export_tensorflow.cc | 237 const auto& array_shape = array.shape(); in ConvertBoolTensorConst() local
261 const auto& array_shape = array.shape(); in ConvertIntTensorConst() local
311 const auto& array_shape = array.shape(); in ConvertComplex64TensorConst() local
|
| D | tooling_util.cc | 525 const Shape& array_shape = array.shape(); in LogArray() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 326 inline tflite::Dims<4> ToRuntimeDims(const tflite::RuntimeShape& array_shape) { in ToRuntimeDims()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 325 const RuntimeShape& array_shape, in AddBiasAndEvalActivationFunction()
|