Home
last modified time | relevance | path

Searched refs:new_operands (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.cc115 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl() argument
117 CHECK_EQ(new_operands.size(), 3); in CloneWithNewOperandsImpl()
119 shape, new_operands[0], new_operands[1], new_operands[2], epsilon(), in CloneWithNewOperandsImpl()
136 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl() argument
138 CHECK_EQ(new_operands.size(), 5); in CloneWithNewOperandsImpl()
140 shape, new_operands[0], new_operands[1], new_operands[2], new_operands[3], in CloneWithNewOperandsImpl()
141 new_operands[4], epsilon(), feature_index()); in CloneWithNewOperandsImpl()
157 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl() argument
159 CHECK_EQ(new_operands.size(), 5); in CloneWithNewOperandsImpl()
161 shape, new_operands[0], new_operands[1], new_operands[2], new_operands[3], in CloneWithNewOperandsImpl()
[all …]
Dhlo_instructions.h70 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
84 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
98 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
124 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
150 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
178 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
203 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
248 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
260 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
272 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
[all …]
Dstable_sort_expander.cc98 std::vector<HloInstruction*> new_operands(sort->operands().begin(), in ExpandInstruction() local
100 new_operands.push_back(iota); in ExpandInstruction()
107 sort->CloneWithNewOperands(new_sort_shape, new_operands)); in ExpandInstruction()
167 std::vector<HloInstruction*> new_operands; in ExpandInstruction() local
168 new_operands.reserve(inst->operand_count()); in ExpandInstruction()
170 new_operands.push_back(replace(operand)); in ExpandInstruction()
173 inst->CloneWithNewOperands(inst->shape(), new_operands); in ExpandInstruction()
Dhlo_element_type_converter.cc172 std::vector<HloInstruction*> new_operands; in Run() local
177 new_operands.push_back(operand); in Run()
190 hlo->CloneWithNewOperands(shape, new_operands, &context)); in Run()
200 hlo->CloneWithNewOperands(new_shape, new_operands, &context)); in Run()
208 hlo->CloneWithNewOperands(hlo->shape(), new_operands, &context)); in Run()
Ddefuser.cc70 std::vector<HloInstruction*> new_operands; in Defuse() local
72 new_operands.push_back(defused_instructions.at(operand)); in Defuse()
77 new_operands)); in Defuse()
Dcall_inliner.cc45 std::vector<HloInstruction*> new_operands; in DefaultAction() local
48 new_operands.push_back(new_operand); in DefaultAction()
51 auto new_hlo = hlo->CloneWithNewOperands(hlo->shape(), new_operands); in DefaultAction()
Dhlo_instruction.cc1334 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperands() argument
1338 for (const HloInstruction* new_operand : new_operands) { in CloneWithNewOperands()
1392 clone = CloneWithNewOperandsImpl(shape, new_operands, context); in CloneWithNewOperands()
1417 CHECK_EQ(new_operands.size(), 1); in CloneWithNewOperands()
1418 clone = CreateUnary(shape, opcode_, new_operands[0]); in CloneWithNewOperands()
1437 CHECK_EQ(new_operands.size(), 2); in CloneWithNewOperands()
1438 clone = CreateBinary(shape, opcode_, new_operands[0], new_operands[1]); in CloneWithNewOperands()
1444 CHECK_EQ(new_operands.size(), 3); in CloneWithNewOperands()
1445 clone = CreateTernary(shape, opcode_, new_operands[0], new_operands[1], in CloneWithNewOperands()
1446 new_operands[2]); in CloneWithNewOperands()
[all …]
Dwhile_loop_invariant_code_motion.cc70 InlinedVector<HloInstruction*, 4> new_operands; in CreateLoopInvariantCopy() local
72 std::back_inserter(new_operands), get_new_operand); in CreateLoopInvariantCopy()
76 old_instruction->shape(), new_operands)); in CreateLoopInvariantCopy()
Dalgebraic_simplifier.cc724 std::vector<HloInstruction*> new_operands; in HandleConcatenate() local
726 new_operands.push_back(operands[k]); in HandleConcatenate()
728 new_operands.push_back(new_slice_op); in HandleConcatenate()
730 new_operands.push_back(operands[k]); in HandleConcatenate()
734 concatenate->shape(), new_operands)); in HandleConcatenate()
2422 std::vector<HloInstruction*> new_operands; in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand() local
2423 new_operands.reserve(user->operand_count()); in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()
2428 new_operands.push_back( in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()
2435 new_operands.push_back(operand); in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()
2445 new_operands)); in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()
[all …]
Dhlo_computation.cc1004 std::vector<HloInstruction*> new_operands; in CloneWithReplacements() local
1010 new_operands.push_back(context->GetInstruction(replaced_operand)); in CloneWithReplacements()
1013 instr->CloneWithNewOperands(instr->shape(), new_operands, context)); in CloneWithReplacements()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dcombine_access_chains.cpp122 std::vector<Operand>* new_operands) { in CombineIndices() argument
165 new_operands->push_back({SPV_OPERAND_TYPE_ID, {new_value_id}}); in CombineIndices()
171 std::vector<Operand>* new_operands) { in CreateNewInputOperands() argument
174 new_operands->push_back(ptr_input->GetInOperand(i)); in CreateNewInputOperands()
181 if (!CombineIndices(ptr_input, inst, new_operands)) return false; in CreateNewInputOperands()
185 new_operands->push_back( in CreateNewInputOperands()
192 new_operands->push_back(inst->GetInOperand(i)); in CreateNewInputOperands()
247 std::vector<Operand> new_operands; in CombineAccessChain() local
248 if (!CreateNewInputOperands(ptr_input, inst, &new_operands)) return false; in CombineAccessChain()
252 inst->SetInOperands(std::move(new_operands)); in CombineAccessChain()
Dlocal_access_chain_convert_pass.cpp86 Instruction::OperandList new_operands; in ReplaceAccessChainLoad() local
89 new_operands.emplace_back(original_load->GetOperand(0)); in ReplaceAccessChainLoad()
90 new_operands.emplace_back(original_load->GetOperand(1)); in ReplaceAccessChainLoad()
92 new_operands.emplace_back( in ReplaceAccessChainLoad()
94 AppendConstantOperands(address_inst, &new_operands); in ReplaceAccessChainLoad()
96 original_load->ReplaceOperands(new_operands); in ReplaceAccessChainLoad()
Dcombine_access_chains.h62 std::vector<Operand>* new_operands);
67 std::vector<Operand>* new_operands);
Dinstruction.h215 inline void SetInOperands(OperandList&& new_operands);
303 void ReplaceOperands(const OperandList& new_operands);
517 inline void Instruction::SetInOperands(OperandList&& new_operands) { in SetInOperands() argument
521 operands_.insert(operands_.end(), new_operands.begin(), new_operands.end()); in SetInOperands()
Dinstruction.cpp141 void Instruction::ReplaceOperands(const OperandList& new_operands) { in ReplaceOperands() argument
143 operands_.insert(operands_.begin(), new_operands.begin(), new_operands.end()); in ReplaceOperands()
Dfolding_rules.cpp2076 std::vector<Operand> new_operands; in VectorShuffleFeedingShuffle() local
2077 new_operands.resize( in VectorShuffleFeedingShuffle()
2128 new_operands.push_back( in VectorShuffleFeedingShuffle()
2151 for (uint32_t i = 2; i < new_operands.size(); i++) { in VectorShuffleFeedingShuffle()
2153 new_operands[i].words[0] -= adjustment; in VectorShuffleFeedingShuffle()
2158 new_operands[0].words[0] = new_feeder_id; in VectorShuffleFeedingShuffle()
2159 new_operands[1] = inst->GetInOperand(1); in VectorShuffleFeedingShuffle()
2161 new_operands[1].words[0] = new_feeder_id; in VectorShuffleFeedingShuffle()
2162 new_operands[0] = inst->GetInOperand(0); in VectorShuffleFeedingShuffle()
2165 inst->SetInOperands(std::move(new_operands)); in VectorShuffleFeedingShuffle()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dcombine_access_chains.cpp122 std::vector<Operand>* new_operands) { in CombineIndices() argument
165 new_operands->push_back({SPV_OPERAND_TYPE_ID, {new_value_id}}); in CombineIndices()
171 std::vector<Operand>* new_operands) { in CreateNewInputOperands() argument
174 new_operands->push_back(ptr_input->GetInOperand(i)); in CreateNewInputOperands()
181 if (!CombineIndices(ptr_input, inst, new_operands)) return false; in CreateNewInputOperands()
185 new_operands->push_back( in CreateNewInputOperands()
192 new_operands->push_back(inst->GetInOperand(i)); in CreateNewInputOperands()
247 std::vector<Operand> new_operands; in CombineAccessChain() local
248 if (!CreateNewInputOperands(ptr_input, inst, &new_operands)) return false; in CombineAccessChain()
252 inst->SetInOperands(std::move(new_operands)); in CombineAccessChain()
Dlocal_access_chain_convert_pass.cpp86 Instruction::OperandList new_operands; in ReplaceAccessChainLoad() local
89 new_operands.emplace_back(original_load->GetOperand(0)); in ReplaceAccessChainLoad()
90 new_operands.emplace_back(original_load->GetOperand(1)); in ReplaceAccessChainLoad()
92 new_operands.emplace_back( in ReplaceAccessChainLoad()
94 AppendConstantOperands(address_inst, &new_operands); in ReplaceAccessChainLoad()
96 original_load->ReplaceOperands(new_operands); in ReplaceAccessChainLoad()
Dcombine_access_chains.h63 std::vector<Operand>* new_operands);
68 std::vector<Operand>* new_operands);
Dinstruction.h215 inline void SetInOperands(OperandList&& new_operands);
303 void ReplaceOperands(const OperandList& new_operands);
517 inline void Instruction::SetInOperands(OperandList&& new_operands) { in SetInOperands() argument
521 operands_.insert(operands_.end(), new_operands.begin(), new_operands.end()); in SetInOperands()
Dinstruction.cpp141 void Instruction::ReplaceOperands(const OperandList& new_operands) { in ReplaceOperands() argument
143 operands_.insert(operands_.begin(), new_operands.begin(), new_operands.end()); in ReplaceOperands()
Dfolding_rules.cpp2076 std::vector<Operand> new_operands; in VectorShuffleFeedingShuffle() local
2077 new_operands.resize( in VectorShuffleFeedingShuffle()
2128 new_operands.push_back( in VectorShuffleFeedingShuffle()
2151 for (uint32_t i = 2; i < new_operands.size(); i++) { in VectorShuffleFeedingShuffle()
2153 new_operands[i].words[0] -= adjustment; in VectorShuffleFeedingShuffle()
2158 new_operands[0].words[0] = new_feeder_id; in VectorShuffleFeedingShuffle()
2159 new_operands[1] = inst->GetInOperand(1); in VectorShuffleFeedingShuffle()
2161 new_operands[1].words[0] = new_feeder_id; in VectorShuffleFeedingShuffle()
2162 new_operands[0] = inst->GetInOperand(0); in VectorShuffleFeedingShuffle()
2165 inst->SetInOperands(std::move(new_operands)); in VectorShuffleFeedingShuffle()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dvariadic_op_splitter.cc43 std::vector<HloInstruction*> new_operands; in SplitConcatenate() local
54 new_operands.insert(new_operands.end(), in SplitConcatenate()
70 new_operands.push_back(new_concat); in SplitConcatenate()
73 operands_to_split = new_operands; in SplitConcatenate()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_extractor.cc70 std::vector<HloInstruction*> new_operands; in DefaultAction() local
72 new_operands.push_back(clone_context_.GetInstruction(operand)); in DefaultAction()
75 hlo->CloneWithNewOperands(hlo->shape(), new_operands, &clone_context_); in DefaultAction()
/external/tensorflow/tensorflow/compiler/jit/
Ddeadness_analysis.cc441 std::vector<Predicate*> new_operands; in SimplifyUsingDeMorgan() local
442 absl::c_transform(pred->GetOperands(), std::back_inserter(new_operands), in SimplifyUsingDeMorgan()
444 return kind == Predicate::Kind::kOr ? MakeAndPredicate(new_operands) in SimplifyUsingDeMorgan()
445 : MakeOrPredicate(new_operands); in SimplifyUsingDeMorgan()

12