Home
last modified time | relevance | path

Searched refs:operand_num (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder.cc89 bool ShouldSkipPadOnOperand(const HloInstruction* inst, int64 operand_num, in ShouldSkipPadOnOperand() argument
93 operand_num == 0 && inst->window().dimensions(dimension).size() == 1) { in ShouldSkipPadOnOperand()
97 if (operand_num == 0 && inst->opcode() == HloOpcode::kConvolution && in ShouldSkipPadOnOperand()
116 for (int64 operand_num = 0; operand_num < inst->operand_count(); in Run() local
117 ++operand_num) { in Run()
118 HloInstruction* operand = inst->mutable_operand(operand_num); in Run()
128 VLOG(1) << "Has dynamic dimension of operand" << operand_num << " @" in Run()
131 if (ShouldSkipPadOnOperand(inst, operand_num, dim)) { in Run()
175 TF_RETURN_IF_ERROR(inst->ReplaceOperandWith(operand_num, padded)); in Run()
176 operand = inst->mutable_operand(operand_num); in Run()
Dhlo_module.cc425 for (int64 operand_num = 0; in OutlineExpressionFromComputation() local
426 operand_num < outlined_instruction->operand_count(); ++operand_num) { in OutlineExpressionFromComputation()
428 outlined_instruction->mutable_operand(operand_num); in OutlineExpressionFromComputation()
442 outlined_instruction->ReplaceOperandWith(operand_num, *operand_slot)); in OutlineExpressionFromComputation()
Dhlo_instructions.cc1415 for (int64 operand_num = 0; operand_num < operand_count(); ++operand_num) { in CloneAndFuseInternal() local
1416 if (instruction_to_fuse == operand(operand_num)) { in CloneAndFuseInternal()
1418 HloInstruction* fused_parameter = fused_parameters[operand_num]; in CloneAndFuseInternal()
1424 fused_instructions_computation()->RemoveParameter(operand_num)); in CloneAndFuseInternal()
1425 RemoveOperandAt(operand_num); in CloneAndFuseInternal()
1448 for (int64 operand_num = 0; operand_num < clone->operand_count(); in CloneAndFuseInternal() local
1449 ++operand_num) { in CloneAndFuseInternal()
1450 HloInstruction* operand = clone->mutable_operand(operand_num); in CloneAndFuseInternal()
1468 TF_CHECK_OK(clone->ReplaceOperandWith(operand_num, fused_param)); in CloneAndFuseInternal()
Dhlo_instruction.cc1517 for (int64 operand_num = 0; operand_num < operand_count(); ++operand_num) { in ~HloInstruction() local
1518 HloInstruction* operand = operands_[operand_num]; in ~HloInstruction()
1525 operands_[operand_num] = nullptr; in ~HloInstruction()
1917 Status HloInstruction::ReplaceOperandWith(int64 operand_num, in ReplaceOperandWith() argument
1919 auto old_operand = operand(operand_num); in ReplaceOperandWith()
1924 return ReplaceOperandWithDifferentShape(operand_num, new_operand); in ReplaceOperandWith()
1928 int64 operand_num, HloInstruction* new_operand) { in ReplaceOperandWithDifferentShape() argument
1929 TF_RET_CHECK(operand_num >= 0); in ReplaceOperandWithDifferentShape()
1930 TF_RET_CHECK(operand_num < operand_count()); in ReplaceOperandWithDifferentShape()
1931 HloInstruction* old_operand = mutable_operand(operand_num); in ReplaceOperandWithDifferentShape()
[all …]
Dpattern_matcher.h2227 inline auto WithOperands(Matcher&& m, int64 operand_num, FirstArg&& first_arg)
2228 -> decltype(m.WithOperand(operand_num, std::forward<FirstArg>(first_arg))) {
2229 return m.WithOperand(operand_num, std::forward<FirstArg>(first_arg));
2233 inline auto WithOperands(Matcher&& m, int64 operand_num, FirstArg&& first_arg,
2235 -> decltype(WithOperands(m.WithOperand(operand_num,
2237 operand_num + 1, std::forward<Args>(args)...)) {
2239 m.WithOperand(operand_num, std::forward<FirstArg>(first_arg)),
2240 operand_num + 1, std::forward<Args>(args)...);
Dhlo_instruction.h975 Status ReplaceOperandWith(int64 operand_num, HloInstruction* new_operand);
978 Status ReplaceOperandWithDifferentShape(int64 operand_num,
Dalgebraic_simplifier.cc2735 int64 operand_num = 0; in TrySimplifyScalarSlice() local
2738 TF_RET_CHECK(operand_num < concat->operand_count()); in TrySimplifyScalarSlice()
2739 const HloInstruction* operand = concat->operand(operand_num); in TrySimplifyScalarSlice()
2745 operand_num++; in TrySimplifyScalarSlice()
2749 slice, concat->mutable_operand(operand_num)); in TrySimplifyScalarSlice()
2756 slice->shape(), concat->mutable_operand(operand_num), in TrySimplifyScalarSlice()
Dhlo_graph_dumper.cc1176 int64 operand_num, bool control_edge = false) { in AddInstructionIncomingEdges() argument
1190 StrFormat(R"( headlabel="%d", labeldistance=2)", operand_num); in AddInstructionIncomingEdges()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dfused_ir_emitter.cc274 for (int64 operand_num = 0; operand_num < consumer->operand_count(); in IsFusedIrEmitterInefficient() local
275 ++operand_num) { in IsFusedIrEmitterInefficient()
276 if (consumer->operand(operand_num) == producer) { in IsFusedIrEmitterInefficient()
277 auto instruction = consumer->fused_parameter(operand_num); in IsFusedIrEmitterInefficient()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc727 if (auto operand_num = in ProfitableToMakeDotOperandColumnMajor() local
729 auto* operand = fusion_root_op->operand(*operand_num); in ProfitableToMakeDotOperandColumnMajor()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1521 int64 operand_num = 0; in CustomCall() local
1527 operand_num); in CustomCall()
1530 ++operand_num; in CustomCall()