Searched refs:input_start_indices (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.cc | 585 HloInstruction* input_start_indices = operands(2); in CreateFromProto() local
594 if (input_start_indices->shape() in CreateFromProto()
613 input_start_indices->shape().tuple_shapes(0))); in CreateFromProto()
618 input_start_indices->shape().tuple_shapes(i)); in CreateFromProto()
635 ShapeUtil::TupleElementCount(input_start_indices->shape())); in CreateFromProto()
636 if (input_start_indices->shape().tuple_shapes(0).IsTuple()) { in CreateFromProto()
638 i < ShapeUtil::TupleElementCount(input_start_indices->shape()); in CreateFromProto()
1334 HloInstruction* input_start_indices, HloInstruction* output_start_indices, in CreateCollectivePermute() argument
1339 HloOpcode::kCollectivePermute, shape, input, output, input_start_indices, in CreateCollectivePermute()
1356 HloInstruction* input_start_indices, HloInstruction* output_start_indices, in CreateCollectivePermuteStart() argument
[all …]
|
| D | hlo_instruction.h | 794 HloInstruction* input_start_indices, HloInstruction* output_start_indices,
808 HloInstruction* input_start_indices, HloInstruction* output_start_indices,
|
| D | hlo_instructions.cc | 999 HloInstruction* output, HloInstruction* input_start_indices, in HloCollectivePermuteInstruction() argument
1010 AppendOperand(input_start_indices); in HloCollectivePermuteInstruction()
|
| D | hlo_instructions.h | 765 HloInstruction* output, HloInstruction* input_start_indices,
|