HomeSort by relevance Sort by last modified time
    Searched refs:IndexIterationSpace (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/
shape_util.h 638 struct IndexIterationSpace {
646 const Shape& shape, const IndexIterationSpace& iteration_space,
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_evaluator.cc 779 // Returns an ShapeUtil::IndexIterationSpace that iterates over the output batch
781 ShapeUtil::IndexIterationSpace IterationSpaceForOutputBatchIndices(
797 // Return an ShapeUtil::IndexIterationSpace that iterates over the output slice
799 ShapeUtil::IndexIterationSpace IterationSpaceForOutputOffsetIndices(
    [all...]
hlo_evaluator_typed_visitor.h     [all...]

Completed in 494 milliseconds