Searched refs:IndexIterationSpace (Results 1 – 2 of 2) sorted by relevance
640 struct IndexIterationSpace { struct648 const Shape& shape, const IndexIterationSpace& iteration_space, in ForEachIndexWithStatus() argument
2196 ShapeUtil::IndexIterationSpace IterationSpaceForOutputBatchIndices( in IterationSpaceForOutputBatchIndices()2214 ShapeUtil::IndexIterationSpace IterationSpaceForOutputOffsetIndices( in IterationSpaceForOutputOffsetIndices()2480 ShapeUtil::IndexIterationSpace start_indices_iteration_space = in HandleGather()2482 ShapeUtil::IndexIterationSpace offset_indices_iteration_space = in HandleGather()2578 ShapeUtil::IndexIterationSpace GetIterationSpaceImpl( in GetIterationSpaceImpl()2607 ShapeUtil::IndexIterationSpace IterationSpaceForUpdateScatterIndices( in IterationSpaceForUpdateScatterIndices()2617 ShapeUtil::IndexIterationSpace IterationSpaceForUpdateWindowIndices( in IterationSpaceForUpdateWindowIndices()2862 ShapeUtil::IndexIterationSpace scatter_indices_iteration_space = in HandleScatter()2864 ShapeUtil::IndexIterationSpace window_indices_iteration_space = in HandleScatter()