Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_propagation.cc32 for (const ShapeUtil::IndexedShape& indexed_shape : in Run() local
43 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
905 for (const auto& indexed_shape : in HandleFusion() local
1105 for (const ShapeUtil::IndexedShape& indexed_shape : in GetBytesRead() local
1123 for (const ShapeUtil::IndexedShape& indexed_shape : in GetBytesWritten() local
Dhlo_dataflow_analysis.cc1229 for (auto& indexed_shape : ShapeUtil::GetLeafShapes(instruction->shape())) { in GetInPlaceInputOutputPairs() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc560 const Shape& indexed_shape = use->operand(0)->shape(); in CreateLiteralForConstrainedUses() local
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc204 for (const ShapeUtil::IndexedShape& indexed_shape : in FromLiterals() local