HomeSort by relevance Sort by last modified time
    Searched defs:multi_index (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/
index_util_test.cc 40 std::vector<int64> multi_index = local
42 EXPECT_EQ(1, multi_index.size());
43 EXPECT_EQ(42, multi_index[0]);
133 std::vector<int64> multi_index = local
136 shape, multi_index));
  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
ir_array.h 104 std::vector<llvm::Value*> multi_index = multidim(); local
105 multi_index[dim] = b->CreateAdd(multi_index[dim], addend);
106 return Index(multi_index, index_type_);
llvm_loop.cc 244 std::vector<llvm::Value*> multi_index(shape.dimensions_size());
251 multi_index[dimension] = loop->GetIndVarValue();
253 return multi_index;
272 std::vector<llvm::Value*> multi_index = local
276 for (size_t dimension = 0; dimension < multi_index.size(); ++dimension) {
278 DCHECK_EQ(nullptr, multi_index[dimension]);
280 DCHECK_NE(nullptr, multi_index[dimension]);
283 return multi_index;
  /external/tensorflow/tensorflow/compiler/xla/service/
elemental_ir_emitter.cc 2054 std::vector<llvm::Value*> multi_index = padded_index.multidim(); local
    [all...]
hlo_parser.cc 2084 auto* multi_index = absl::get_if<absl::Span<const int64>>(&index); local
    [all...]

Completed in 389 milliseconds