| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | fused_ir_emitter.cc | 279 for (int64 operand_num = 0; operand_num < consumer->operand_count(); in IsFusedIrEmitterInefficient() local
280 ++operand_num) { in IsFusedIrEmitterInefficient()
281 if (consumer->operand(operand_num) == producer) { in IsFusedIrEmitterInefficient()
282 auto instruction = consumer->fused_parameter(operand_num); in IsFusedIrEmitterInefficient()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_padder.cc | 126 bool ShouldSkipPadOnOperand(const HloInstruction* inst, int64 operand_num, in ShouldSkipPadOnOperand() argument
130 operand_num == 0 && inst->window().dimensions(dimension).size() == 1) { in ShouldSkipPadOnOperand()
134 if (operand_num == 0 && inst->opcode() == HloOpcode::kConvolution && in ShouldSkipPadOnOperand()
1023 for (int64 operand_num = 0; operand_num < inst->operand_count(); in Run() local
1024 ++operand_num) { in Run()
1025 HloInstruction* original_operand = inst->mutable_operand(operand_num); in Run()
1044 VLOG(2) << "Has dynamic dimension of operand" << operand_num << " @" in Run()
1047 if (ShouldSkipPadOnOperand(inst, operand_num, input_dim)) { in Run()
1052 ChooseIdentityValue(inst, operand_num)); in Run()
1059 TF_RETURN_IF_ERROR(inst->ReplaceOperandWith(operand_num, padded)); in Run()
[all …]
|
| D | hlo_cost_analysis.h | 157 int64 operand_bytes_accessed(const HloInstruction& hlo, int64 operand_num,
211 void SetOperandBytesAccessed(int64 operand_num, float value);
212 void SetOperandBytesAccessed(int64 operand_num, ShapeIndex index,
221 static std::string GetOperandBytesAccessedKey(int64 operand_num,
|
| D | hlo_cost_analysis.cc | 1002 int64 operand_num, in operand_bytes_accessed() argument
1004 return GetPropertyForHlo(hlo, GetOperandBytesAccessedKey(operand_num, index), in operand_bytes_accessed()
1033 void HloCostAnalysis::SetOperandBytesAccessed(int64 operand_num, float value) { in SetOperandBytesAccessed() argument
1034 current_properties_[GetOperandBytesAccessedKey(operand_num).c_str()] = value; in SetOperandBytesAccessed()
1037 void HloCostAnalysis::SetOperandBytesAccessed(int64 operand_num, in SetOperandBytesAccessed() argument
1039 current_properties_[GetOperandBytesAccessedKey(operand_num, index).c_str()] = in SetOperandBytesAccessed()
1052 int64 operand_num, ShapeIndex index) { in GetOperandBytesAccessedKey() argument
1053 return absl::StrCat(kBytesAccessedKey, " operand ", operand_num, " ", in GetOperandBytesAccessedKey()
|
| D | hlo_module.cc | 481 for (int64 operand_num = 0; in OutlineExpressionFromComputation() local
482 operand_num < outlined_instruction->operand_count(); ++operand_num) { in OutlineExpressionFromComputation()
484 outlined_instruction->mutable_operand(operand_num); in OutlineExpressionFromComputation()
498 outlined_instruction->ReplaceOperandWith(operand_num, *operand_slot)); in OutlineExpressionFromComputation()
|
| D | hlo_instruction.cc | 1639 for (int64 operand_num = 0; operand_num < operand_count(); ++operand_num) { in ~HloInstruction() local
1640 HloInstruction* operand = operands_[operand_num]; in ~HloInstruction()
1647 operands_[operand_num] = nullptr; in ~HloInstruction()
2058 Status HloInstruction::ReplaceOperandWith(int64 operand_num, in ReplaceOperandWith() argument
2060 auto old_operand = operand(operand_num); in ReplaceOperandWith()
2065 return ReplaceOperandWithDifferentShape(operand_num, new_operand); in ReplaceOperandWith()
2069 int64 operand_num, HloInstruction* new_operand) { in ReplaceOperandWithDifferentShape() argument
2070 TF_RET_CHECK(operand_num >= 0); in ReplaceOperandWithDifferentShape()
2071 TF_RET_CHECK(operand_num < operand_count()); in ReplaceOperandWithDifferentShape()
2072 HloInstruction* old_operand = mutable_operand(operand_num); in ReplaceOperandWithDifferentShape()
[all …]
|
| D | hlo_instructions.cc | 1539 for (int64 operand_num = 0; operand_num < operand_count(); ++operand_num) { in CloneAndFuseInternal() local
1540 if (instruction_to_fuse == operand(operand_num)) { in CloneAndFuseInternal()
1542 HloInstruction* fused_parameter = fused_parameters[operand_num]; in CloneAndFuseInternal()
1548 fused_instructions_computation()->RemoveParameter(operand_num)); in CloneAndFuseInternal()
1549 RemoveOperandAt(operand_num); in CloneAndFuseInternal()
1572 for (int64 operand_num = 0; operand_num < clone->operand_count(); in CloneAndFuseInternal() local
1573 ++operand_num) { in CloneAndFuseInternal()
1574 HloInstruction* operand = clone->mutable_operand(operand_num); in CloneAndFuseInternal()
1592 TF_CHECK_OK(clone->ReplaceOperandWith(operand_num, fused_param)); in CloneAndFuseInternal()
|
| D | pattern_matcher.h | 2240 inline auto WithOperands(Matcher&& m, int64 operand_num, FirstArg&& first_arg)
2241 -> decltype(m.WithOperand(operand_num, std::forward<FirstArg>(first_arg))) {
2242 return m.WithOperand(operand_num, std::forward<FirstArg>(first_arg));
2246 inline auto WithOperands(Matcher&& m, int64 operand_num, FirstArg&& first_arg,
2248 -> decltype(WithOperands(m.WithOperand(operand_num,
2250 operand_num + 1, std::forward<Args>(args)...)) {
2252 m.WithOperand(operand_num, std::forward<FirstArg>(first_arg)),
2253 operand_num + 1, std::forward<Args>(args)...);
|
| D | hlo_instruction.h | 1131 Status ReplaceOperandWith(int64 operand_num, HloInstruction* new_operand);
1134 Status ReplaceOperandWithDifferentShape(int64 operand_num,
1926 UseKind OperandElementUse(int64 operand_num) const;
|
| D | hlo_graph_dumper.cc | 1194 int64 operand_num, bool control_edge = false) { in AddInstructionIncomingEdges() argument
1209 StrFormat(R"( headlabel="%d", labeldistance=2)", operand_num); in AddInstructionIncomingEdges()
|
| D | algebraic_simplifier.cc | 3264 int64 operand_num = 0; in TrySimplifyScalarSlice() local
3267 TF_RET_CHECK(operand_num < concat->operand_count()); in TrySimplifyScalarSlice()
3268 const HloInstruction* operand = concat->operand(operand_num); in TrySimplifyScalarSlice()
3274 operand_num++; in TrySimplifyScalarSlice()
3278 slice, concat->mutable_operand(operand_num)); in TrySimplifyScalarSlice()
3285 slice->shape(), concat->mutable_operand(operand_num), in TrySimplifyScalarSlice()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_driver.cc | 640 int operand_num = use.getOperandNumber(); in PreprocessConstantOps() local
644 if (biases.find(operand_num) == biases.end() && in PreprocessConstantOps()
647 auto it = spec->coeff_op_quant_dim.find(operand_num); in PreprocessConstantOps()
652 bias_users.push_back({user, operand_num}); in PreprocessConstantOps()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 777 if (auto operand_num = in ProfitableToMakeDotOperandColumnMajor() local
779 auto* operand = fusion_root_op->operand(*operand_num); in ProfitableToMakeDotOperandColumnMajor()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1577 int64 operand_num = 0; in CustomCall() local
1583 operand_num); in CustomCall()
1586 ++operand_num; in CustomCall()
|