Searched refs:array_shape (Results 1 - 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util_test.cc378 Shape array_shape = ShapeUtil::MakeShape(F32, {42, 42, 123}); local
380 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(array_shape, {})));
382 array_shape, *ShapeUtil::GetMutableSubshape(&array_shape, {})));
386 ShapeUtil::MakeTupleShape({array_shape, array_shape, array_shape});
390 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(tuple_shape, {0})));
392 ShapeUtil::Equal(array_shape, ShapeUti
416 Shape array_shape = ShapeUtil::MakeShape(F32, {42, 42, 123}); local
[all...]
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
H A Dresolve_constant_concatenation.cc90 const Shape array_shape = input_array->shape(); local
92 for (int i = concatenation_axis; i < array_shape.dimensions_count(); i++) {
93 array_copy_size[count] *= array_shape.dims()[i];
/external/tensorflow/tensorflow/contrib/lite/toco/
H A Dmodel.h1553 bool has_shape() const { return array_shape != nullptr; }
1556 return *array_shape;
1559 if (!array_shape) {
1560 array_shape.reset(new Shape);
1562 return array_shape.get();
1565 void clear_shape() { array_shape = nullptr; }
1607 std::unique_ptr<Shape> array_shape; member in struct:toco::Array
H A Ddump_graphviz.cc155 auto& array_shape = array.shape(); local
157 for (int id = 0; id < array_shape.dimensions_count(); id++) {
159 AppendF(&node_properties.label, "%d", array_shape.dims(id));
162 AppendF(&node_properties.label, "\u00D7%d", array_shape.dims(id));
H A Dtooling_util.cc415 const Shape& array_shape = array.shape(); local
416 if (array_shape.dimensions_count() == 0) {
421 for (const int dim : array_shape.dims()) {
H A Dexport_tensorflow.cc234 const auto& array_shape = array.shape(); local
236 for (int i = 0; i < array_shape.dimensions_count(); i++) {
237 shape->add_dim()->set_size(array_shape.dims(i));
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dlocal_client_execute_test.cc291 const Shape array_shape = ShapeUtil::MakeShape(F32, {2, 2}); local
295 ShapeUtil::MakeTupleShape({array_shape, vector_shape});
297 ShapeUtil::MakeTupleShape({vector_shape, array_shape});
338 const Shape array_shape = ShapeUtil::MakeShape(F32, {2, 2}); local
342 ShapeUtil::MakeTupleShape({array_shape, vector_shape});
383 const Shape array_shape = ShapeUtil::MakeShape(F32, {2, 2}); local
385 ShapeUtil::MakeTupleShape({array_shape, array_shape});

Completed in 470 milliseconds