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

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_matchers.h 62 tuple_index_(tuple_index) {}
68 int64 tuple_index_; member in class:xla::testing::HloGetTupleElementMatcher
hlo_instruction.h 1376 int64 tuple_index_ = -1; member in class:xla::HloInstruction
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
while_transformer.cc 248 int64 tuple_index() const { return tuple_index_; }
284 // Get tagged GTE instruction and set 'tuple_index_'.
287 tuple_index_ = gte->tuple_index();
324 int64 tuple_index_ = -1; member in class:xla::gpu::__anon38929::WhileConditionComputationMatcher
338 : while_hlo_(while_hlo), tuple_index_(tuple_index) {
358 ExprTree(HloOpcode::kTuple, tuple_index_,
386 const int64 tuple_index_; member in class:xla::gpu::__anon38929::WhileInitOperandMatcher
403 : computation_(computation), tuple_index_(tuple_index) {
439 ExprTree tuple0(HloOpcode::kTuple, tuple_index_,
454 if (tag == "gte" && inst->tuple_index() != tuple_index_) {
484 const int64 tuple_index_; member in class:xla::gpu::__anon38929::WhileBodyComputationMatcher
    [all...]

Completed in 178 milliseconds