Home
last modified time | relevance | path

Searched refs:array_shape (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc439 Shape array_shape = ShapeUtil::MakeShape(F32, {42, 42, 123}); in TEST() local
441 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(array_shape, {}))); in TEST()
443 array_shape, *ShapeUtil::GetMutableSubshape(&array_shape, {}))); in TEST()
447 ShapeUtil::MakeTupleShape({array_shape, array_shape, array_shape}); in TEST()
451 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(tuple_shape, {0}))); in TEST()
453 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(tuple_shape, {1}))); in TEST()
455 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(tuple_shape, {2}))); in TEST()
459 {array_shape, ShapeUtil::MakeTupleShape({array_shape, array_shape}), in TEST()
461 {ShapeUtil::MakeTupleShape({array_shape, array_shape}), in TEST()
462 array_shape})}); in TEST()
[all …]
Dshape_test.cc87 Shape array_shape = in TEST_F() local
89 EXPECT_EQ("f32[<=23,44,<=55]", array_shape.ToString()); in TEST_F()
91 array_shape.set_dynamic_dimension(2, false); in TEST_F()
92 EXPECT_EQ("f32[<=23,44,55]", array_shape.ToString()); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshaped_buffer_test.cc155 Shape array_shape = ShapeUtil::MakeShape(F32, {1}); in TEST() local
157 xla::ShapeUtil::MakeTupleShape({array_shape, array_shape}); in TEST()
170 EXPECT_EQ(ssb.on_host_shape(), array_shape); in TEST()
171 EXPECT_EQ(ssb.on_device_shape(), array_shape); in TEST()
Dpattern_matcher_test.cc87 auto array_shape = ShapeUtil::MakeShape(F32, {2, 3, 4}); in TEST_F() local
89 EXPECT_TRUE(Match(&array_shape, match::Shape(&matched_shape).IsArray())); in TEST_F()
90 EXPECT_EQ(matched_shape, &array_shape); in TEST_F()
91 EXPECT_TRUE(Match(&array_shape, match::Shape().IsDenseArray())); in TEST_F()
92 EXPECT_FALSE(Match(&array_shape, match::Shape().IsScalar())); in TEST_F()
93 EXPECT_FALSE(Match(&array_shape, match::Shape().IsTuple())); in TEST_F()
94 EXPECT_TRUE(Match(&array_shape, match::Shape().WithElementType(F32))); in TEST_F()
95 EXPECT_TRUE(Match(&array_shape, match::Shape().WithRank(3))); in TEST_F()
97 Match(&array_shape, match::Shape().WithSubshape({0}, match::Shape()))); in TEST_F()
99 EXPECT_TRUE(Match(&array_shape, in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client_backend_independent_test.py42 xla_client.Shape.array_shape(xla_client.PrimitiveType.F32, [-2, 4])
46 xla_client.Shape.array_shape(xla_client.PrimitiveType.F32, [2, 4], [3])
50 xla_client.Shape.array_shape(xla_client.PrimitiveType.F32, [2, 4],
Dxla_client_test.py215 shape = xla_client.Shape.array_shape(
335 shape_with_layout=xla_client.Shape.array_shape(
338 xla_client.Shape.array_shape(np.dtype(np.float32), (), ()),
339 xla_client.Shape.array_shape(np.dtype(np.float32), (), ()),
1212 shape=xla_client.Shape.array_shape(xla_client.PrimitiveType.F32,
1228 shape=xla_client.Shape.array_shape(xla_client.PrimitiveType.F32,
1246 shape=xla_client.Shape.array_shape(xla_client.PrimitiveType.S32,
1874 xla_client.Shape.array_shape(np.dtype(np.float32), [])
Dxla_client.py289 return Shape.array_shape(pyval.dtype, np.shape(pyval))
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_concatenation.cc91 const Shape array_shape = input_array->shape(); in ConcatenateTensorBuffers() local
93 for (int i = concatenation_axis; i < array_shape.dimensions_count(); i++) { in ConcatenateTensorBuffers()
94 array_copy_size[count] *= array_shape.dims()[i]; in ConcatenateTensorBuffers()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
Dfuse_binary_into_preceding_affine_test.cc49 Shape* array_shape = array.mutable_shape(); in CreateArray() local
50 *(array_shape->mutable_dims()) = shape; in CreateArray()
Dfuse_binary_into_following_affine_test.cc49 Shape* array_shape = array.mutable_shape(); in CreateArray() local
50 *(array_shape->mutable_dims()) = shape; in CreateArray()
Dremove_successive_transpose_test.cc37 toco::Shape* array_shape = array.mutable_shape(); in CreateArray() local
38 *(array_shape->mutable_dims()) = shape; in CreateArray()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_execute_test.cc282 const Shape array_shape = ShapeUtil::MakeShape(F32, {2, 2}); in XLA_TEST_F() local
286 ShapeUtil::MakeTupleShape({array_shape, vector_shape}); in XLA_TEST_F()
288 ShapeUtil::MakeTupleShape({vector_shape, array_shape}); in XLA_TEST_F()
328 const Shape array_shape = ShapeUtil::MakeShape(F32, {2, 2}); in XLA_TEST_F() local
332 ShapeUtil::MakeTupleShape({array_shape, vector_shape}); in XLA_TEST_F()
371 const Shape array_shape = ShapeUtil::MakeShape(F32, {2, 2}); in XLA_TEST_F() local
373 ShapeUtil::MakeTupleShape({array_shape, array_shape}); in XLA_TEST_F()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h2302 bool has_shape() const { return array_shape != nullptr; }
2305 return *array_shape;
2308 if (!array_shape) {
2309 array_shape.reset(new Shape);
2311 return array_shape.get();
2314 void clear_shape() { array_shape = nullptr; }
2390 std::unique_ptr<Shape> array_shape;
Ddump_graphviz.cc282 auto& array_shape = array.shape(); in GetArrayLabel() local
284 for (int dim = 0; dim < array_shape.dimensions_count(); dim++) { in GetArrayLabel()
285 AppendF(&html, "%d", array_shape.dims(dim)); in GetArrayLabel()
286 if (dim + 1 < array_shape.dimensions_count()) { in GetArrayLabel()
Dexport_tensorflow.cc237 const auto& array_shape = array.shape(); in ConvertBoolTensorConst() local
239 for (int i = 0; i < array_shape.dimensions_count(); i++) { in ConvertBoolTensorConst()
240 shape->add_dim()->set_size(array_shape.dims(i)); in ConvertBoolTensorConst()
261 const auto& array_shape = array.shape(); in ConvertIntTensorConst() local
263 for (int i = 0; i < array_shape.dimensions_count(); i++) { in ConvertIntTensorConst()
264 shape->add_dim()->set_size(array_shape.dims(i)); in ConvertIntTensorConst()
311 const auto& array_shape = array.shape(); in ConvertComplex64TensorConst() local
313 for (int i = 0; i < array_shape.dimensions_count(); i++) { in ConvertComplex64TensorConst()
314 shape->add_dim()->set_size(array_shape.dims(i)); in ConvertComplex64TensorConst()
Dtooling_util.cc525 const Shape& array_shape = array.shape(); in LogArray() local
526 if (array_shape.dimensions_count() == 0) { in LogArray()
531 for (const int dim : array_shape.dims()) { in LogArray()
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddriver.cc439 ArrayShape array_shape = shape.elements[tuple_idx]; in Display() local
440 Display(casted[tuple_idx], array_shape); in Display()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h326 inline tflite::Dims<4> ToRuntimeDims(const tflite::RuntimeShape& array_shape) { in ToRuntimeDims() argument
328 const int dimensions_count = array_shape.DimensionsCount(); in ToRuntimeDims()
333 (i < dimensions_count) ? array_shape.Dims(dimensions_count - 1 - i) : 1; in ToRuntimeDims()
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dtypes.h489 inline tflite::Dims<4> ToRuntimeDims(const tflite::RuntimeShape& array_shape) {
491 const int dimensions_count = array_shape.DimensionsCount();
496 (i < dimensions_count) ? array_shape.Dims(dimensions_count - 1 - i) : 1;
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h325 const RuntimeShape& array_shape, in AddBiasAndEvalActivationFunction() argument
328 bias_shape.FlatSize(), bias_data, array_shape.FlatSize(), in AddBiasAndEvalActivationFunction()