Home
last modified time | relevance | path

Searched defs:array_shape (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
Dfuse_binary_into_following_affine_test.cc49 Shape* array_shape = array.mutable_shape(); in CreateArray() local
/external/tensorflow/tensorflow/compiler/xla/
Dshape_test.cc84 Shape array_shape = in TEST_F() local
Dshape_util_test.cc415 Shape array_shape = ShapeUtil::MakeShape(F32, {42, 42, 123}); in TEST() local
453 Shape array_shape = ShapeUtil::MakeShape(F32, {42, 42, 123}); in TEST() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_concatenation.cc91 const Shape array_shape = input_array->shape(); in ConcatenateTensorBuffers() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dpattern_matcher_test.cc84 auto array_shape = ShapeUtil::MakeShape(F32, {2, 3, 4}); in TEST() local
107 auto array_shape = ShapeUtil::MakeShapeWithSparseLayout(F32, {2, 3, 4}, 10); in TEST() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_execute_test.cc279 const Shape array_shape = ShapeUtil::MakeShape(F32, {2, 2}); in XLA_TEST_F() local
325 const Shape array_shape = ShapeUtil::MakeShape(F32, {2, 2}); in XLA_TEST_F() local
368 const Shape array_shape = ShapeUtil::MakeShape(F32, {2, 2}); in XLA_TEST_F() local
/external/tensorflow/tensorflow/lite/toco/
Ddump_graphviz.cc279 auto& array_shape = array.shape(); in GetArrayLabel() local
Dexport_tensorflow.cc234 const auto& array_shape = array.shape(); in ConvertBoolTensorConst() local
258 const auto& array_shape = array.shape(); in ConvertIntTensorConst() local
307 const auto& array_shape = array.shape(); in ConvertComplex64TensorConst() local
Dtooling_util.cc497 const Shape& array_shape = array.shape(); in LogArray() local
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h280 inline tflite::Dims<4> ToRuntimeDims(const tflite::RuntimeShape& array_shape) { in ToRuntimeDims()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py486 def array_shape(element_type, dimensions, minor_to_major=None): member in Shape
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h296 inline tflite::Dims<4> ToRuntimeDims(const tflite::RuntimeShape& array_shape) { in ToRuntimeDims()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h191 const RuntimeShape& array_shape, in AddBiasAndEvalActivationFunction()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h286 const RuntimeShape& array_shape, in AddBiasAndEvalActivationFunction()