Home
last modified time | relevance | path

Searched defs:indexed_shape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_propagation.cc37 for (const ShapeUtil::IndexedShape& indexed_shape : in Run() local
48 for (const ShapeUtil::IndexedShape& indexed_shape : in Run() local
Dhlo_cost_analysis.cc342 for (const auto& indexed_shape : ShapeUtil::GetLeafShapes(infeed->shape())) { in HandleInfeed() local
358 for (const auto& indexed_shape : in HandleOutfeed() local
934 for (const auto& indexed_shape : in HandleFusion() local
1134 for (const ShapeUtil::IndexedShape& indexed_shape : in GetBytesRead() local
1152 for (const ShapeUtil::IndexedShape& indexed_shape : in GetBytesWritten() local
Dhlo_dataflow_analysis.cc1613 for (auto& indexed_shape : ShapeUtil::GetLeafShapes(instruction->shape())) { in GetInPlaceInputOutputPairs() local
Dmemory_space_assignment.cc1981 for (const auto& indexed_shape : in AddInputAndOutputRequiredAssignments() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc562 const Shape& indexed_shape = use->operand(0)->shape(); in CreateLiteralForConstrainedUses() local
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc205 for (const ShapeUtil::IndexedShape& indexed_shape : in FromLiterals() local