| /frameworks/ml/nn/tools/test_generator/include/ |
| TestHarness.h | 119 template <typename ty, size_t tuple_index>
123 auto& t = std::get<tuple_index>(test);
135 template <typename ty, size_t tuple_index>
140 auto& g = std::get<tuple_index>(*filtered);
158 template <size_t tuple_index>
161 std::function<void(VALUE_TYPE(tuple_index), VALUE_TYPE(tuple_index))>
163 for_each<VALUE_TYPE(tuple_index)>(
165 [&test, &cmp](int index, const VECTOR_TYPE(tuple_index) & m) {
166 const auto& test_operands = std::get<tuple_index>(test)
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| tuple_simplifier.cc | 77 operand->tuple_index() != operand_number) {
114 instruction->tuple_index());
|
| hlo_matchers.cc | 97 if (instruction->tuple_index() != tuple_index_) {
98 *listener << "has wrong tuple index (got " << instruction->tuple_index()
|
| hlo_matchers.h | 60 int64 tuple_index)
62 tuple_index_(tuple_index) {}
190 ::testing::Matcher<const HloInstruction*> operand, int64 tuple_index) {
192 new ::xla::testing::HloGetTupleElementMatcher(operand, tuple_index));
|
| while_loop_simplifier.cc | 96 if (tuple_idx && tuple_idx != operand->tuple_index()) {
98 << *tuple_idx << " vs " << operand->tuple_index();
102 tuple_idx = operand->tuple_index();
354 while_body_root->operand_index(user) == user->tuple_index() &&
360 used_tuple_indices.insert(user->tuple_index());
380 operand->tuple_index() != i) {
445 int64 old_idx = user->tuple_index();
|
| hlo.proto | 64 int64 tuple_index = 13;
|
| tuple_points_to_analysis_test.cc | 672 // Check that Gte at tuple_index = 0 points-to fusion_param({0})
677 // Check that Gte at tuple_index = 1 points-to fusion_param({1})
682 // Check that Gte at tuple_index = 2 points-to fusion_param({2})
[all...] |
| hlo_tfgraph_builder.cc | 186 attrs["index"].set_i(instruction->tuple_index());
|
| while_loop_invariant_code_motion.cc | 137 instr->tuple_index() == i &&
|
| hlo_computation_test.cc | 283 EXPECT_EQ(0, tuple_copy->operand(0)->operand(0)->tuple_index());
284 EXPECT_EQ(1, tuple_copy->operand(1)->operand(0)->tuple_index());
|
| hlo_instruction.cc | 109 instruction->tuple_index_ = proto.tuple_index();
1605 int64 HloInstruction::tuple_index() const { function in class:xla::HloInstruction
[all...] |
| tuple_points_to_analysis.cc | 226 int64 element_index = get_tuple_element->tuple_index();
|
| hlo_dataflow_analysis.cc | 410 ShapeIndex operand_index = {gte->tuple_index()};
[all...] |
| hlo_instruction.h | 712 int64 tuple_index() const;
[all...] |
| hlo_verifier.cc | 154 get_tuple_element->tuple_index()));
[all...] |
| layout_assignment.cc | [all...] |
| algebraic_simplifier.cc | [all...] |
| hlo_evaluator.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| while_transformer.cc | 56 // tuple_index, ExprTree(HloOpcode::kParameter)));
66 // constants or recording the tuple_index).
230 // *) 'tuple_index':
231 // *) The loop induction variable tuple_index from the GetTupleElement
248 int64 tuple_index() const { return tuple_index_; } function in class:xla::gpu::__anon38929::WhileConditionComputationMatcher
287 tuple_index_ = gte->tuple_index();
330 // while instructions 'init' operand, indexing the tuple at 'tuple_index'.
337 const int64 tuple_index)
338 : while_hlo_(while_hlo), tuple_index_(tuple_index) {
395 // computation root[tuple_index] to param[tuple_index], where 'tuple_index
[all...] |
| while_transformer_test.cc | 42 const int64 tuple_index, const int64 limit) {
47 0, GetLoopStateShape(tuple_index), "loop_state"));
50 limit_const->shape(), loop_state, tuple_index));
|
| hlo_to_ir_bindings.cc | 131 gte->shape(), gte->tuple_index(), /*alignment=*/1,
135 gte->shape(), gte->tuple_index(), /*alignment=*/1,
|
| ir_emitter.cc | 118 get_tuple_element->shape(), get_tuple_element->tuple_index(),
[all...] |
| ir_emitter_unnested.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| fused_ir_emitter.cc | 104 get_tuple_element->shape(), get_tuple_element->tuple_index(),
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| ir_emitter.cc | 308 shape, get_tuple_element->tuple_index(), MinimumAlignmentForShape(shape),
[all...] |
