Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dconditional_code_motion.cc313 for (int64 operand_num = 0; operand_num < old_root->operand_count(); in FindSpecialConverts() local
314 ++operand_num) { in FindSpecialConverts()
315 if (old_root->operand(operand_num)->opcode() != HloOpcode::kConvert) { in FindSpecialConverts()
320 old_root->mutable_operand(operand_num); in FindSpecialConverts()
327 ? ShapeUtil::Equal(others_root->operand(operand_num)->shape(), in FindSpecialConverts()
330 others_root->operand(operand_num)->shape(), in FindSpecialConverts()
332 if ((others_root->operand(operand_num)->opcode() == in FindSpecialConverts()
342 kspecial_convert.insert(operand_num); in FindSpecialConverts()
441 for (int64 operand_num = 0; operand_num < old_root->operand_count(); in ConvertSpecialMove() local
442 ++operand_num) { in ConvertSpecialMove()
[all …]
Dhlo_cost_analysis.h162 int64 operand_bytes_accessed(const HloInstruction& hlo, int64 operand_num,
188 static std::string GetOperandBytesAccessedKey(int64 operand_num,
230 void SetOperandBytesAccessed(int64 operand_num, float value);
231 void SetOperandBytesAccessed(int64 operand_num, ShapeIndex index,
Dhlo_cost_analysis.cc1084 int64 operand_num, in operand_bytes_accessed() argument
1086 return GetPropertyForHlo(hlo, GetOperandBytesAccessedKey(operand_num, index), in operand_bytes_accessed()
1151 void HloCostAnalysis::SetOperandBytesAccessed(int64 operand_num, float value) { in SetOperandBytesAccessed() argument
1152 current_properties_[GetOperandBytesAccessedKey(operand_num).c_str()] = value; in SetOperandBytesAccessed()
1155 void HloCostAnalysis::SetOperandBytesAccessed(int64 operand_num, in SetOperandBytesAccessed() argument
1157 current_properties_[GetOperandBytesAccessedKey(operand_num, index).c_str()] = in SetOperandBytesAccessed()
1170 int64 operand_num, ShapeIndex index) { in GetOperandBytesAccessedKey() argument
1171 return absl::StrCat(kBytesAccessedKey, " operand ", operand_num, " ", in GetOperandBytesAccessedKey()
Ddynamic_padder.cc221 bool ShouldSkipPadOnOperand(const HloInstruction* inst, int64 operand_num, in ShouldSkipPadOnOperand() argument
225 operand_num == 0 && inst->window().dimensions(dimension).size() == 1) { in ShouldSkipPadOnOperand()
229 if (operand_num == 0 && inst->opcode() == HloOpcode::kConvolution && in ShouldSkipPadOnOperand()
1885 for (int64 operand_num = 0; operand_num < inst->operand_count(); in Run() local
1886 ++operand_num) { in Run()
1887 HloInstruction* original_operand = inst->mutable_operand(operand_num); in Run()
1901 VLOG(2) << "Has dynamic dimension of operand" << operand_num << " @" in Run()
1904 if (ShouldSkipPadOnOperand(inst, operand_num, input_dim)) { in Run()
1909 ChooseIdentityValue(inst, operand_num)); in Run()
1916 TF_RETURN_IF_ERROR(inst->ReplaceOperandWith(operand_num, padded)); in Run()
[all …]
Dhlo_module.cc526 for (int64 operand_num = 0; in OutlineExpressionFromComputation() local
527 operand_num < outlined_instruction->operand_count(); ++operand_num) { in OutlineExpressionFromComputation()
529 outlined_instruction->mutable_operand(operand_num); in OutlineExpressionFromComputation()
543 outlined_instruction->ReplaceOperandWith(operand_num, *operand_slot)); in OutlineExpressionFromComputation()
Dhlo_instruction.cc1846 for (int64 operand_num = 0; operand_num < operand_count(); ++operand_num) { in DetachFromOperandsAndUsers() local
1847 HloInstruction* operand = operands_[operand_num]; in DetachFromOperandsAndUsers()
1854 operands_[operand_num] = nullptr; in DetachFromOperandsAndUsers()
2329 Status HloInstruction::ReplaceOperandWith(int64 operand_num, in ReplaceOperandWith() argument
2331 auto old_operand = operand(operand_num); in ReplaceOperandWith()
2336 return ReplaceOperandWithDifferentShape(operand_num, new_operand); in ReplaceOperandWith()
2340 int64 operand_num, HloInstruction* new_operand) { in ReplaceOperandWithDifferentShape() argument
2341 TF_RET_CHECK(operand_num >= 0); in ReplaceOperandWithDifferentShape()
2342 TF_RET_CHECK(operand_num < operand_count()); in ReplaceOperandWithDifferentShape()
2343 HloInstruction* old_operand = mutable_operand(operand_num); in ReplaceOperandWithDifferentShape()
[all …]
Dhlo_instructions.cc1684 for (int64 operand_num = 0; operand_num < operand_count(); ++operand_num) { in CloneAndFuseInternal() local
1685 if (instruction_to_fuse == operand(operand_num)) { in CloneAndFuseInternal()
1687 HloInstruction* fused_parameter = fused_parameters[operand_num]; in CloneAndFuseInternal()
1693 fused_instructions_computation()->RemoveParameter(operand_num)); in CloneAndFuseInternal()
1694 RemoveOperandAt(operand_num); in CloneAndFuseInternal()
1717 for (int64 operand_num = 0; operand_num < clone->operand_count(); in CloneAndFuseInternal() local
1718 ++operand_num) { in CloneAndFuseInternal()
1719 HloInstruction* operand = clone->mutable_operand(operand_num); in CloneAndFuseInternal()
1737 TF_CHECK_OK(clone->ReplaceOperandWith(operand_num, fused_param)); in CloneAndFuseInternal()
Dpattern_matcher.h2133 inline auto WithOperands(Matcher&& m, int64 operand_num, FirstArg&& first_arg) {
2134 return m.WithOperand(operand_num, std::forward<FirstArg>(first_arg));
2138 inline auto WithOperands(Matcher&& m, int64 operand_num, FirstArg&& first_arg,
2141 m.WithOperand(operand_num, std::forward<FirstArg>(first_arg)),
2142 operand_num + 1, std::forward<Args>(args)...);
Dhlo_instruction.h1197 Status ReplaceOperandWith(int64 operand_num, HloInstruction* new_operand);
1200 Status ReplaceOperandWithDifferentShape(int64 operand_num,
2049 UseKind OperandElementUse(int64 operand_num) const;
Dhlo_graph_dumper.cc1214 int64 operand_num, bool control_edge = false) { in AddInstructionIncomingEdges() argument
1229 StrFormat(R"( headlabel="%d", labeldistance=2)", operand_num); in AddInstructionIncomingEdges()
Dalgebraic_simplifier.cc4006 int64 operand_num = 0; in TrySimplifyScalarSlice() local
4009 TF_RET_CHECK(operand_num < concat->operand_count()); in TrySimplifyScalarSlice()
4010 const HloInstruction* operand = concat->operand(operand_num); in TrySimplifyScalarSlice()
4016 operand_num++; in TrySimplifyScalarSlice()
4020 slice, concat->mutable_operand(operand_num)); in TrySimplifyScalarSlice()
4027 slice->shape(), concat->mutable_operand(operand_num), in TrySimplifyScalarSlice()
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_driver.cc689 int operand_num = indexed_use.value().second; in PreprocessConstantOps() local
698 if (biases.find(operand_num) == biases.end() && in PreprocessConstantOps()
707 if (affine_user && affine_user.GetAffineOperandIndex() == operand_num && in PreprocessConstantOps()
720 user->setOperand(operand_num, new_cst); in PreprocessConstantOps()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc914 if (auto operand_num = in ProfitableToMakeDotOperandColumnMajor() local
916 auto* operand = fusion_root_op->operand(*operand_num); in ProfitableToMakeDotOperandColumnMajor()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1906 int64 operand_num = 0; in CustomCall() local
1912 operand_num); in CustomCall()
1914 ++operand_num; in CustomCall()
1993 int64 operand_num = 0; in CustomCall() local
1999 operand_num); in CustomCall()
2002 ++operand_num; in CustomCall()