Searched refs:array_shape (Results 1 - 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/ |
H A D | 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, ShapeUti 416 Shape array_shape = ShapeUtil::MakeShape(F32, {42, 42, 123}); local [all...] |
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
H A D | 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/ |
H A D | model.h | 1553 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 D | 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));
|
H A D | 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()) {
|
H A D | 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));
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
H A D | 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});
|
Completed in 470 milliseconds