Home
last modified time | relevance | path

Searched defs:linear_index (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dindex_util.cc76 int64 linear_index = 0; in MultidimensionalIndexToLinearIndex() local
93 const Shape& shape, int64 linear_index) { in LinearIndexToMultidimensionalIndex()
Dindex_util_test.cc132 for (auto linear_index : linear_indexes) { in TEST() local
Dliteral_comparison.cc286 int64 linear_index; member
372 void CompareValues(T expected, T actual, int64 linear_index) { in CompareValues()
461 void CompareValues(complex64 expected, complex64 actual, int64 linear_index) { in CompareValues()
484 int64 linear_index) { in CompareValues()
Dliteral.cc215 auto linear_index = [](const Shape& shape, in CopySliceFromInternal() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dparallel_loop_emitter.cc125 llvm::Value* linear_index = in EmitIndexAndSetExitBasicBlock() local
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_tiling.cc63 llvm::Value* linear_index = index.GetConstantWithIndexType(0); in GetReshapedIndex() local
Dir_array.cc232 llvm::Value* linear_index = GetConstantWithIndexType(0); in SourceIndexOfBitcast() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser.cc2073 if (auto* linear_index = absl::get_if<int64>(&index)) { in SetValueInLiteralHelper() local
2179 int64 linear_index = 0; in ParseDenseLiteral() local