OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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