Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc437 Shape array_shape = ShapeUtil::MakeShape(F32, {42, 42, 123}); in TEST() local
439 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(array_shape, {}))); in TEST()
441 array_shape, *ShapeUtil::GetMutableSubshape(&array_shape, {}))); in TEST()
445 ShapeUtil::MakeTupleShape({array_shape, array_shape, array_shape}); in TEST()
449 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(tuple_shape, {0}))); in TEST()
451 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(tuple_shape, {1}))); in TEST()
453 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(tuple_shape, {2}))); in TEST()
457 {array_shape, ShapeUtil::MakeTupleShape({array_shape, array_shape}), in TEST()
459 {ShapeUtil::MakeTupleShape({array_shape, array_shape}), in TEST()
460 array_shape})}); in TEST()
[all …]
Dshape_test.cc86 Shape array_shape = in TEST_F() local
88 EXPECT_EQ("f32[<=23,44,<=55]", array_shape.ToString()); in TEST_F()
90 array_shape.set_dynamic_dimension(2, false); in TEST_F()
91 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()
171 EXPECT_EQ(ssb.on_host_shape(), array_shape); in TEST()
172 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/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_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.h2291 bool has_shape() const { return array_shape != nullptr; }
2294 return *array_shape;
2297 if (!array_shape) {
2298 array_shape.reset(new Shape);
2300 return array_shape.get();
2303 void clear_shape() { array_shape = nullptr; }
2379 std::unique_ptr<Shape> array_shape;
Ddump_graphviz.cc279 auto& array_shape = array.shape(); in GetArrayLabel() local
281 for (int dim = 0; dim < array_shape.dimensions_count(); dim++) { in GetArrayLabel()
282 AppendF(&html, "%d", array_shape.dims(dim)); in GetArrayLabel()
283 if (dim + 1 < array_shape.dimensions_count()) { in GetArrayLabel()
Dexport_tensorflow.cc234 const auto& array_shape = array.shape(); in ConvertBoolTensorConst() local
236 for (int i = 0; i < array_shape.dimensions_count(); i++) { in ConvertBoolTensorConst()
237 shape->add_dim()->set_size(array_shape.dims(i)); in ConvertBoolTensorConst()
258 const auto& array_shape = array.shape(); in ConvertIntTensorConst() local
260 for (int i = 0; i < array_shape.dimensions_count(); i++) { in ConvertIntTensorConst()
261 shape->add_dim()->set_size(array_shape.dims(i)); in ConvertIntTensorConst()
307 const auto& array_shape = array.shape(); in ConvertComplex64TensorConst() local
309 for (int i = 0; i < array_shape.dimensions_count(); i++) { in ConvertComplex64TensorConst()
310 shape->add_dim()->set_size(array_shape.dims(i)); in ConvertComplex64TensorConst()
Dtooling_util.cc522 const Shape& array_shape = array.shape(); in LogArray() local
523 if (array_shape.dimensions_count() == 0) { in LogArray()
528 for (const int dim : array_shape.dims()) { in LogArray()
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddriver.cc419 ArrayShape array_shape = shape.elements[tuple_idx]; in Display() local
420 Display(casted[tuple_idx], array_shape); in Display()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py442 return Shape.array_shape(pyval.dtype, np.shape(pyval))
951 xla_shape = _xla.Shape.array_shape(element_type, shape, None)
1306 shape = _xla.Shape.array_shape(self.GetShape(mu).xla_element_type(), dims)
1323 shape = _xla.Shape.array_shape(self.GetShape(a).xla_element_type(), dims)
Dxla_client_test.py345 shape_with_layout=xla_client.Shape.array_shape(
348 xla_client.Shape.array_shape(np.dtype(np.float32), (), ()),
349 xla_client.Shape.array_shape(np.dtype(np.float32), (), ()),
1996 xla_client.Shape.array_shape(np.dtype(np.float32), [])
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h302 inline tflite::Dims<4> ToRuntimeDims(const tflite::RuntimeShape& array_shape) { in ToRuntimeDims() argument
304 const int dimensions_count = array_shape.DimensionsCount(); in ToRuntimeDims()
309 (i < dimensions_count) ? array_shape.Dims(dimensions_count - 1 - i) : 1; in ToRuntimeDims()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h267 const RuntimeShape& array_shape, in AddBiasAndEvalActivationFunction() argument
270 bias_shape.FlatSize(), bias_data, array_shape.FlatSize(), in AddBiasAndEvalActivationFunction()