Home
last modified time | relevance | path

Searched refs:normalized_shape_index (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_tiling.cc150 const IrArray::Index& normalized_shape_index, in GetUnnormalizedIndex() argument
152 DCHECK_EQ(normalized_shape_index.size(), dims_in_elems_.size()); in GetUnnormalizedIndex()
155 return GetReshapedIndex(normalized_shape_index, output_shape, in GetUnnormalizedIndex()
Dkernel_tiling.h175 const IrArray::Index& normalized_shape_index,