| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
| H A D | fuse_binary_into_preceding_affine_test.cc | 50 Shape* array_shape = array.mutable_shape(); in CreateArray() local
|
| H A D | remove_successive_transpose_test.cc | 38 toco::Shape* array_shape = array.mutable_shape(); in CreateArray() local
|
| H A D | fuse_binary_into_following_affine_test.cc | 50 Shape* array_shape = array.mutable_shape(); in CreateArray() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | resolve_constant_concatenation.cc | 92 const Shape array_shape = input_array->shape(); in ConcatenateTensorBuffers() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | shape_test.cc | 87 Shape array_shape = in TEST_F() local
|
| H A D | shape_util_test.cc | 454 Shape array_shape = ShapeUtil::MakeShape(F32, {42, 42, 123}); in TEST() local
492 Shape array_shape = ShapeUtil::MakeShape(F32, {42, 42, 123}); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | shaped_buffer_test.cc | 158 Shape array_shape = ShapeUtil::MakeShape(F32, {1}); in TEST() local
|
| H A D | pattern_matcher_test.cc | 86 auto array_shape = ShapeUtil::MakeShape(F32, {2, 3, 4}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | runtime_shape.h | 191 inline tflite::Dims<4> ToRuntimeDims(const tflite::RuntimeShape& array_shape) { in ToRuntimeDims()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/ |
| H A D | export_tensorflow.cc | 242 const auto& array_shape = array.shape(); in ConvertBoolTensorConst() local
266 const auto& array_shape = array.shape(); in ConvertIntTensorConst() local
316 const auto& array_shape = array.shape(); in ConvertComplex64TensorConst() local
|
| H A D | dump_graphviz.cc | 284 auto& array_shape = array.shape(); in GetArrayLabel() local
|
| H A D | tooling_util.cc | 527 const Shape& array_shape = array.shape(); in LogArray() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/ |
| H A D | driver.cc | 458 ArrayShape array_shape = shape.elements[tuple_idx]; in Display() local
|
| /aosp_15_r20/external/rust/android-crates-io/crates/pdl-compiler/src/backends/rust/ |
| D | decoder.rs | 380 let array_shape = if let Some(count) = size { in add_array_field() localVariable
|
| /aosp_15_r20/external/rust/android-crates-io/crates/pdl-compiler/src/backends/rust_legacy/ |
| D | parser.rs | 366 let array_shape = if let Some(count) = size { in add_array_field() localVariable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | optimized_ops.h | 261 const RuntimeShape& array_shape, in AddBiasAndEvalActivationFunction()
|