| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_util_test.cc | 378 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, ShapeUtil::GetSubshape(tuple_shape, {1})))
416 Shape array_shape = ShapeUtil::MakeShape(F32, {42, 42, 123}); local
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| resolve_constant_concatenation.cc | 90 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/ |
| model.h | 1607 std::unique_ptr<Shape> array_shape; member in struct:toco::Array
[all...] |
| dump_graphviz.cc | 155 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));
|
| tooling_util.cc | 415 const Shape& array_shape = array.shape(); local
416 if (array_shape.dimensions_count() == 0) {
421 for (const int dim : array_shape.dims()) {
[all...] |
| export_tensorflow.cc | 234 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));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| local_client_execute_test.cc | 291 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});
[all...] |
