Searched refs:indexed_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_propagation.cc | 32 for (const ShapeUtil::IndexedShape& indexed_shape : in Run() local
35 int64 memory_space = indexed_shape.shape.layout().memory_space(); in Run()
36 modified |= Propagate(indexed_shape.index, in Run()
43 for (const ShapeUtil::IndexedShape& indexed_shape : in Run() local
45 int64 memory_space = indexed_shape.shape.layout().memory_space(); in Run()
47 Propagate(indexed_shape.index, in Run()
|
| D | hlo_cost_analysis.cc | 342 for (const auto& indexed_shape : ShapeUtil::GetLeafShapes(infeed->shape())) { in HandleInfeed() local
343 size += GetShapeSize(indexed_shape.shape); in HandleInfeed()
344 SetOutputBytesAccessed(indexed_shape.index, in HandleInfeed()
345 GetShapeSize(indexed_shape.shape)); in HandleInfeed()
358 for (const auto& indexed_shape : in HandleOutfeed() local
360 size += GetShapeSize(indexed_shape.shape); in HandleOutfeed()
361 SetOperandBytesAccessed(i, indexed_shape.index, in HandleOutfeed()
362 GetShapeSize(indexed_shape.shape)); in HandleOutfeed()
905 for (const auto& indexed_shape : in HandleFusion() local
908 for (int64 index : indexed_shape.index) { in HandleFusion()
[all …]
|
| D | hlo_dataflow_analysis.cc | 1229 for (auto& indexed_shape : ShapeUtil::GetLeafShapes(instruction->shape())) { in GetInPlaceInputOutputPairs() local
1232 for (int64 i = 0; i < indexed_shape.index.size(); ++i) { in GetInPlaceInputOutputPairs()
1235 hlo_generating_output->operand(indexed_shape.index[i]); in GetInPlaceInputOutputPairs()
1237 CHECK_EQ(i, indexed_shape.index.size() - 1); in GetInPlaceInputOutputPairs()
1254 indexed_shape.index); in GetInPlaceInputOutputPairs()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.cc | 204 for (const ShapeUtil::IndexedShape& indexed_shape : in FromLiterals() local
211 indexed_shape.shape, in FromLiterals()
212 [driver, &leaf, &indexed_shape](tpu_driver::BufferHandle* handle) { in FromLiterals()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 560 const Shape& indexed_shape = use->operand(0)->shape(); in CreateLiteralForConstrainedUses() local
571 ShapeUtil::GetDimension(indexed_shape, operand - first_index) - in CreateLiteralForConstrainedUses()
|