Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.cc118 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl() argument
120 CHECK_EQ(new_operands.size(), 3); in CloneWithNewOperandsImpl()
122 shape, new_operands[0], new_operands[1], new_operands[2], epsilon(), in CloneWithNewOperandsImpl()
139 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl() argument
141 CHECK_EQ(new_operands.size(), 5); in CloneWithNewOperandsImpl()
143 shape, new_operands[0], new_operands[1], new_operands[2], new_operands[3], in CloneWithNewOperandsImpl()
144 new_operands[4], epsilon(), feature_index()); in CloneWithNewOperandsImpl()
160 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl() argument
162 CHECK_EQ(new_operands.size(), 5); in CloneWithNewOperandsImpl()
164 shape, new_operands[0], new_operands[1], new_operands[2], new_operands[3], in CloneWithNewOperandsImpl()
[all …]
Dhlo_instructions.h72 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
86 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
100 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
126 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
152 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
176 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
204 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
229 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
307 const Shape& shape, absl::Span<HloInstruction* const> new_operands,
319 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()
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()
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()
/external/deqp-deps/SPIRV-Tools/source/opt/
Deliminate_dead_members_pass.cpp345 Instruction::OperandList new_operands; in UpdateOpTypeStruct() local
347 new_operands.emplace_back(inst->GetInOperand(idx)); in UpdateOpTypeStruct()
350 inst->SetInOperands(std::move(new_operands)); in UpdateOpTypeStruct()
386 Instruction::OperandList new_operands; in UpdateOpGroupMemberDecorate() local
387 new_operands.emplace_back(inst->GetInOperand(0)); in UpdateOpGroupMemberDecorate()
398 new_operands.emplace_back(inst->GetOperand(i)); in UpdateOpGroupMemberDecorate()
400 new_operands.emplace_back( in UpdateOpGroupMemberDecorate()
404 new_operands.emplace_back(inst->GetOperand(i + 1)); in UpdateOpGroupMemberDecorate()
412 if (new_operands.size() == 1) { in UpdateOpGroupMemberDecorate()
417 inst->SetInOperands(std::move(new_operands)); in UpdateOpGroupMemberDecorate()
[all …]
Damd_ext_to_khr.cpp82 Instruction::OperandList new_operands; in ReplaceTrinaryMinMax() local
83 new_operands.push_back({SPV_OPERAND_TYPE_ID, {glsl405_ext_inst_id}}); in ReplaceTrinaryMinMax()
84 new_operands.push_back({SPV_OPERAND_TYPE_EXTENSION_INSTRUCTION_NUMBER, in ReplaceTrinaryMinMax()
86 new_operands.push_back({SPV_OPERAND_TYPE_ID, {temp->result_id()}}); in ReplaceTrinaryMinMax()
87 new_operands.push_back({SPV_OPERAND_TYPE_ID, {op3}}); in ReplaceTrinaryMinMax()
89 inst->SetInOperands(std::move(new_operands)); in ReplaceTrinaryMinMax()
123 Instruction::OperandList new_operands; in ReplaceTrinaryMid() local
124 new_operands.push_back({SPV_OPERAND_TYPE_ID, {glsl405_ext_inst_id}}); in ReplaceTrinaryMid()
125 new_operands.push_back({SPV_OPERAND_TYPE_EXTENSION_INSTRUCTION_NUMBER, in ReplaceTrinaryMid()
127 new_operands.push_back({SPV_OPERAND_TYPE_ID, {op1}}); in ReplaceTrinaryMid()
[all …]
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()
Dprivate_to_local_pass.cpp72 std::vector<Operand> new_operands; in Process() local
77 new_operands.push_back(entry.GetInOperand(i)); in Process()
80 if (new_operands.size() != entry.NumInOperands()) { in Process()
81 entry.SetInOperands(std::move(new_operands)); in Process()
Dlocal_access_chain_convert_pass.cpp106 Instruction::OperandList new_operands; in ReplaceAccessChainLoad() local
109 new_operands.emplace_back(original_load->GetOperand(0)); in ReplaceAccessChainLoad()
110 new_operands.emplace_back(original_load->GetOperand(1)); in ReplaceAccessChainLoad()
112 new_operands.emplace_back( in ReplaceAccessChainLoad()
114 AppendConstantOperands(address_inst, &new_operands); in ReplaceAccessChainLoad()
116 original_load->ReplaceOperands(new_operands); in ReplaceAccessChainLoad()
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Deliminate_dead_members_pass.cpp345 Instruction::OperandList new_operands; in UpdateOpTypeStruct() local
347 new_operands.emplace_back(inst->GetInOperand(idx)); in UpdateOpTypeStruct()
350 inst->SetInOperands(std::move(new_operands)); in UpdateOpTypeStruct()
386 Instruction::OperandList new_operands; in UpdateOpGroupMemberDecorate() local
387 new_operands.emplace_back(inst->GetInOperand(0)); in UpdateOpGroupMemberDecorate()
398 new_operands.emplace_back(inst->GetOperand(i)); in UpdateOpGroupMemberDecorate()
400 new_operands.emplace_back( in UpdateOpGroupMemberDecorate()
404 new_operands.emplace_back(inst->GetOperand(i + 1)); in UpdateOpGroupMemberDecorate()
412 if (new_operands.size() == 1) { in UpdateOpGroupMemberDecorate()
417 inst->SetInOperands(std::move(new_operands)); in UpdateOpGroupMemberDecorate()
[all …]
Damd_ext_to_khr.cpp82 Instruction::OperandList new_operands; in ReplaceTrinaryMinMax() local
83 new_operands.push_back({SPV_OPERAND_TYPE_ID, {glsl405_ext_inst_id}}); in ReplaceTrinaryMinMax()
84 new_operands.push_back({SPV_OPERAND_TYPE_EXTENSION_INSTRUCTION_NUMBER, in ReplaceTrinaryMinMax()
86 new_operands.push_back({SPV_OPERAND_TYPE_ID, {temp->result_id()}}); in ReplaceTrinaryMinMax()
87 new_operands.push_back({SPV_OPERAND_TYPE_ID, {op3}}); in ReplaceTrinaryMinMax()
89 inst->SetInOperands(std::move(new_operands)); in ReplaceTrinaryMinMax()
123 Instruction::OperandList new_operands; in ReplaceTrinaryMid() local
124 new_operands.push_back({SPV_OPERAND_TYPE_ID, {glsl405_ext_inst_id}}); in ReplaceTrinaryMid()
125 new_operands.push_back({SPV_OPERAND_TYPE_EXTENSION_INSTRUCTION_NUMBER, in ReplaceTrinaryMid()
127 new_operands.push_back({SPV_OPERAND_TYPE_ID, {op1}}); in ReplaceTrinaryMid()
[all …]
Dinterp_fixup_pass.cpp68 Instruction::OperandList new_operands; in ReplaceInternalInterpolate() local
69 new_operands.push_back({SPV_OPERAND_TYPE_ID, {glsl450_ext_inst_id}}); in ReplaceInternalInterpolate()
70 new_operands.push_back( in ReplaceInternalInterpolate()
72 new_operands.push_back({SPV_OPERAND_TYPE_ID, {ptr_id}}); in ReplaceInternalInterpolate()
73 if (op2_id != 0) new_operands.push_back({SPV_OPERAND_TYPE_ID, {op2_id}}); in ReplaceInternalInterpolate()
75 inst->SetInOperands(std::move(new_operands)); in ReplaceInternalInterpolate()
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()
Dprivate_to_local_pass.cpp72 std::vector<Operand> new_operands; in Process() local
77 new_operands.push_back(entry.GetInOperand(i)); in Process()
80 if (new_operands.size() != entry.NumInOperands()) { in Process()
81 entry.SetInOperands(std::move(new_operands)); in Process()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Deliminate_dead_members_pass.cpp345 Instruction::OperandList new_operands; in UpdateOpTypeStruct() local
347 new_operands.emplace_back(inst->GetInOperand(idx)); in UpdateOpTypeStruct()
350 inst->SetInOperands(std::move(new_operands)); in UpdateOpTypeStruct()
386 Instruction::OperandList new_operands; in UpdateOpGroupMemberDecorate() local
387 new_operands.emplace_back(inst->GetInOperand(0)); in UpdateOpGroupMemberDecorate()
398 new_operands.emplace_back(inst->GetOperand(i)); in UpdateOpGroupMemberDecorate()
400 new_operands.emplace_back( in UpdateOpGroupMemberDecorate()
404 new_operands.emplace_back(inst->GetOperand(i + 1)); in UpdateOpGroupMemberDecorate()
412 if (new_operands.size() == 1) { in UpdateOpGroupMemberDecorate()
417 inst->SetInOperands(std::move(new_operands)); in UpdateOpGroupMemberDecorate()
[all …]
Damd_ext_to_khr.cpp82 Instruction::OperandList new_operands; in ReplaceTrinaryMinMax() local
83 new_operands.push_back({SPV_OPERAND_TYPE_ID, {glsl405_ext_inst_id}}); in ReplaceTrinaryMinMax()
84 new_operands.push_back({SPV_OPERAND_TYPE_EXTENSION_INSTRUCTION_NUMBER, in ReplaceTrinaryMinMax()
86 new_operands.push_back({SPV_OPERAND_TYPE_ID, {temp->result_id()}}); in ReplaceTrinaryMinMax()
87 new_operands.push_back({SPV_OPERAND_TYPE_ID, {op3}}); in ReplaceTrinaryMinMax()
89 inst->SetInOperands(std::move(new_operands)); in ReplaceTrinaryMinMax()
123 Instruction::OperandList new_operands; in ReplaceTrinaryMid() local
124 new_operands.push_back({SPV_OPERAND_TYPE_ID, {glsl405_ext_inst_id}}); in ReplaceTrinaryMid()
125 new_operands.push_back({SPV_OPERAND_TYPE_EXTENSION_INSTRUCTION_NUMBER, in ReplaceTrinaryMid()
127 new_operands.push_back({SPV_OPERAND_TYPE_ID, {op1}}); in ReplaceTrinaryMid()
[all …]
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()
Dprivate_to_local_pass.cpp72 std::vector<Operand> new_operands; in Process() local
77 new_operands.push_back(entry.GetInOperand(i)); in Process()
80 if (new_operands.size() != entry.NumInOperands()) { in Process()
81 entry.SetInOperands(std::move(new_operands)); in Process()
Dlocal_access_chain_convert_pass.cpp106 Instruction::OperandList new_operands; in ReplaceAccessChainLoad() local
109 new_operands.emplace_back(original_load->GetOperand(0)); in ReplaceAccessChainLoad()
110 new_operands.emplace_back(original_load->GetOperand(1)); in ReplaceAccessChainLoad()
112 new_operands.emplace_back( in ReplaceAccessChainLoad()
114 AppendConstantOperands(address_inst, &new_operands); in ReplaceAccessChainLoad()
116 original_load->ReplaceOperands(new_operands); in ReplaceAccessChainLoad()
/external/swiftshader/third_party/SPIRV-Tools/source/fuzz/
Dtransformation_set_loop_control.cpp117 opt::Instruction::OperandList new_operands; in Apply() local
119 new_operands.push_back(merge_inst->GetInOperand(0)); in Apply()
120 new_operands.push_back(merge_inst->GetInOperand(1)); in Apply()
122 new_operands.push_back( in Apply()
142 new_operands.push_back( in Apply()
156 new_operands.push_back( in Apply()
162 new_operands.push_back( in Apply()
168 merge_inst->SetInOperands(std::move(new_operands)); in Apply()
/external/deqp-deps/SPIRV-Tools/source/fuzz/
Dtransformation_set_loop_control.cpp117 opt::Instruction::OperandList new_operands; in Apply() local
119 new_operands.push_back(merge_inst->GetInOperand(0)); in Apply()
120 new_operands.push_back(merge_inst->GetInOperand(1)); in Apply()
122 new_operands.push_back( in Apply()
142 new_operands.push_back( in Apply()
156 new_operands.push_back( in Apply()
162 new_operands.push_back( in Apply()
168 merge_inst->SetInOperands(std::move(new_operands)); in Apply()
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/fuzz/
Dtransformation_set_loop_control.cpp119 opt::Instruction::OperandList new_operands; in Apply() local
121 new_operands.push_back(merge_inst->GetInOperand(0)); in Apply()
122 new_operands.push_back(merge_inst->GetInOperand(1)); in Apply()
124 new_operands.push_back( in Apply()
144 new_operands.push_back( in Apply()
158 new_operands.push_back( in Apply()
164 new_operands.push_back( in Apply()
170 merge_inst->SetInOperands(std::move(new_operands)); in Apply()
/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/mlir/lite/transforms/
Dlegalize_tf_while.cc59 SmallVector<Value, 4> new_operands; in RunOnWhile() local
61 new_operands.push_back(block->addArgument(t)); in RunOnWhile()
62 auto call = builder.create<CallOp>(while_op.getLoc(), func, new_operands); in RunOnWhile()

1234