/external/tensorflow/tensorflow/compiler/xla/service/ |
D | conditional_code_motion.cc | 313 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 …]
|
D | hlo_cost_analysis.h | 162 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,
|
D | hlo_cost_analysis.cc | 1084 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()
|
D | dynamic_padder.cc | 221 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 …]
|
D | hlo_module.cc | 526 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()
|
D | hlo_instruction.cc | 1846 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 …]
|
D | hlo_instructions.cc | 1684 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()
|
D | pattern_matcher.h | 2133 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)...);
|
D | hlo_instruction.h | 1197 Status ReplaceOperandWith(int64 operand_num, HloInstruction* new_operand); 1200 Status ReplaceOperandWithDifferentShape(int64 operand_num, 2049 UseKind OperandElementUse(int64 operand_num) const;
|
D | hlo_graph_dumper.cc | 1214 int64 operand_num, bool control_edge = false) { in AddInstructionIncomingEdges() argument 1229 StrFormat(R"( headlabel="%d", labeldistance=2)", operand_num); in AddInstructionIncomingEdges()
|
D | algebraic_simplifier.cc | 4006 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/ |
D | quantization_driver.cc | 689 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/ |
D | dot_op_emitter.cc | 914 if (auto operand_num = in ProfitableToMakeDotOperandColumnMajor() local 916 auto* operand = fusion_root_op->operand(*operand_num); in ProfitableToMakeDotOperandColumnMajor()
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 1906 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()
|