Home
last modified time | relevance | path

Searched refs:operand_index (Results 1 – 25 of 175) sorted by relevance

1234567

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_support.cc24 int64_t operand_index) const { in SupportsBF16Operand()
36 CHECK_EQ(operand_index, 0); in SupportsBF16Operand()
82 const HloInstruction& hlo, int64_t operand_index) { in EffectiveOperandPrecisionIsOutputPrecision() argument
110 return operand_index == 0; in EffectiveOperandPrecisionIsOutputPrecision()
112 return operand_index == 0 || operand_index == 1; in EffectiveOperandPrecisionIsOutputPrecision()
114 return operand_index == 0; in EffectiveOperandPrecisionIsOutputPrecision()
116 return operand_index == 1 || operand_index == 2; in EffectiveOperandPrecisionIsOutputPrecision()
139 const HloInstruction& hlo, int64_t operand_index) const { in EffectiveOperandPrecisionIsBF16()
Ddynamic_dimension_inference.cc174 int64_t operand_index, HloInstruction* dynamic_size)>;
180 int64_t operand_index,
185 int64_t operand_index,
189 int64_t operand_index,
194 int64_t operand_index,
200 int64_t operand_index,
235 int64_t operand_index, HloInstruction* dynamic_size) { in DefaultAction() argument
247 int64_t operand_index, HloInstruction* dynamic_size) { in HandleGetTupleElement() argument
259 int64_t operand_index, HloInstruction* dynamic_size) { in HandleTuple() argument
260 index.push_front(operand_index); in HandleTuple()
[all …]
Dhlo_liveness_analysis.cc119 for (int64_t operand_index = 0; operand_index < instruction->operand_count(); in PropagateLivenessThroughTuple() local
120 ++operand_index) { in PropagateLivenessThroughTuple()
123 if (shape_index.empty() || shape_index[0] != operand_index) { in PropagateLivenessThroughTuple()
127 MarkLiveAtIndex(instruction->operand(operand_index), {}, live_index_map, in PropagateLivenessThroughTuple()
134 MarkLiveAtIndex(instruction->operand(operand_index), operand_shape_index, in PropagateLivenessThroughTuple()
245 int64_t operand_index = 1; in PropagateLivenessThroughControlFlow() local
248 MarkLiveAtIndex(caller->operand(operand_index), {}, live_index_map, in PropagateLivenessThroughControlFlow()
256 MarkLiveAtIndex(caller->operand(operand_index), shape_index, in PropagateLivenessThroughControlFlow()
262 ++operand_index; in PropagateLivenessThroughControlFlow()
/external/deqp-deps/SPIRV-Tools/source/val/
Dvalidate_arithmetics.cpp51 for (size_t operand_index = 2; operand_index < inst->operands().size(); in ArithmeticsPass() local
52 ++operand_index) { in ArithmeticsPass()
53 if (_.GetOperandTypeId(inst, operand_index) != result_type) in ArithmeticsPass()
57 << operand_index; in ArithmeticsPass()
73 for (size_t operand_index = 2; operand_index < inst->operands().size(); in ArithmeticsPass() local
74 ++operand_index) { in ArithmeticsPass()
75 if (_.GetOperandTypeId(inst, operand_index) != result_type) in ArithmeticsPass()
79 << operand_index; in ArithmeticsPass()
102 for (size_t operand_index = 2; operand_index < inst->operands().size(); in ArithmeticsPass() local
103 ++operand_index) { in ArithmeticsPass()
[all …]
Dvalidate_composites.cpp227 for (uint32_t operand_index = 2; operand_index < num_operands; in ValidateCompositeConstruct() local
228 ++operand_index) { in ValidateCompositeConstruct()
229 const uint32_t operand_type = _.GetOperandTypeId(inst, operand_index); in ValidateCompositeConstruct()
268 for (uint32_t operand_index = 2; operand_index < num_operands; in ValidateCompositeConstruct() local
269 ++operand_index) { in ValidateCompositeConstruct()
270 const uint32_t operand_type = _.GetOperandTypeId(inst, operand_index); in ValidateCompositeConstruct()
303 for (uint32_t operand_index = 2; operand_index < num_operands; in ValidateCompositeConstruct() local
304 ++operand_index) { in ValidateCompositeConstruct()
305 const uint32_t operand_type = _.GetOperandTypeId(inst, operand_index); in ValidateCompositeConstruct()
326 for (uint32_t operand_index = 2; operand_index < num_operands; in ValidateCompositeConstruct() local
[all …]
Dvalidate_bitwise.cpp118 for (size_t operand_index = 2; operand_index < inst->operands().size(); in BitwisePass() local
119 ++operand_index) { in BitwisePass()
120 const uint32_t type_id = _.GetOperandTypeId(inst, operand_index); in BitwisePass()
126 << operand_index; in BitwisePass()
132 << " operand index " << operand_index; in BitwisePass()
138 << " operand index " << operand_index; in BitwisePass()
/external/swiftshader/third_party/SPIRV-Tools/source/val/
Dvalidate_arithmetics.cpp52 for (size_t operand_index = 2; operand_index < inst->operands().size(); in ArithmeticsPass() local
53 ++operand_index) { in ArithmeticsPass()
54 if (_.GetOperandTypeId(inst, operand_index) != result_type) in ArithmeticsPass()
58 << operand_index; in ArithmeticsPass()
74 for (size_t operand_index = 2; operand_index < inst->operands().size(); in ArithmeticsPass() local
75 ++operand_index) { in ArithmeticsPass()
76 if (_.GetOperandTypeId(inst, operand_index) != result_type) in ArithmeticsPass()
80 << operand_index; in ArithmeticsPass()
104 for (size_t operand_index = 2; operand_index < inst->operands().size(); in ArithmeticsPass() local
105 ++operand_index) { in ArithmeticsPass()
[all …]
Dvalidate_composites.cpp228 for (uint32_t operand_index = 2; operand_index < num_operands; in ValidateCompositeConstruct() local
229 ++operand_index) { in ValidateCompositeConstruct()
230 const uint32_t operand_type = _.GetOperandTypeId(inst, operand_index); in ValidateCompositeConstruct()
269 for (uint32_t operand_index = 2; operand_index < num_operands; in ValidateCompositeConstruct() local
270 ++operand_index) { in ValidateCompositeConstruct()
271 const uint32_t operand_type = _.GetOperandTypeId(inst, operand_index); in ValidateCompositeConstruct()
304 for (uint32_t operand_index = 2; operand_index < num_operands; in ValidateCompositeConstruct() local
305 ++operand_index) { in ValidateCompositeConstruct()
306 const uint32_t operand_type = _.GetOperandTypeId(inst, operand_index); in ValidateCompositeConstruct()
327 for (uint32_t operand_index = 2; operand_index < num_operands; in ValidateCompositeConstruct() local
[all …]
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/val/
Dvalidate_arithmetics.cpp53 for (size_t operand_index = 2; operand_index < inst->operands().size(); in ArithmeticsPass() local
54 ++operand_index) { in ArithmeticsPass()
56 const uint32_t type_id = _.GetOperandTypeId(inst, operand_index); in ArithmeticsPass()
62 << operand_index; in ArithmeticsPass()
67 } else if (_.GetOperandTypeId(inst, operand_index) != result_type) in ArithmeticsPass()
71 << operand_index; in ArithmeticsPass()
87 for (size_t operand_index = 2; operand_index < inst->operands().size(); in ArithmeticsPass() local
88 ++operand_index) { in ArithmeticsPass()
90 const uint32_t type_id = _.GetOperandTypeId(inst, operand_index); in ArithmeticsPass()
96 << operand_index; in ArithmeticsPass()
[all …]
Dvalidate_composites.cpp227 for (uint32_t operand_index = 2; operand_index < num_operands; in ValidateCompositeConstruct() local
228 ++operand_index) { in ValidateCompositeConstruct()
229 const uint32_t operand_type = _.GetOperandTypeId(inst, operand_index); in ValidateCompositeConstruct()
268 for (uint32_t operand_index = 2; operand_index < num_operands; in ValidateCompositeConstruct() local
269 ++operand_index) { in ValidateCompositeConstruct()
270 const uint32_t operand_type = _.GetOperandTypeId(inst, operand_index); in ValidateCompositeConstruct()
303 for (uint32_t operand_index = 2; operand_index < num_operands; in ValidateCompositeConstruct() local
304 ++operand_index) { in ValidateCompositeConstruct()
305 const uint32_t operand_type = _.GetOperandTypeId(inst, operand_index); in ValidateCompositeConstruct()
326 for (uint32_t operand_index = 2; operand_index < num_operands; in ValidateCompositeConstruct() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion.cc59 HloInstruction* consumer, int64_t operand_index) { in ShouldFuseInexpensiveChecks() argument
60 HloInstruction* producer = consumer->mutable_operand(operand_index); in ShouldFuseInexpensiveChecks()
69 ReusesOperandElements(consumer, operand_index)) { in ShouldFuseInexpensiveChecks()
78 !InstructionFusion::ShouldFuse(consumer, operand_index)) { in ShouldFuseInexpensiveChecks()
85 int64_t operand_index) { in ShouldFuse() argument
87 !ShouldFuseInexpensiveChecks(consumer, operand_index)) { in ShouldFuse()
91 auto producer = consumer->operand(operand_index); in ShouldFuse()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Ddef_use_manager.h36 uint32_t operand_index; // logical operand index of the id use. This can be member
41 return lhs.inst == rhs.inst && lhs.operand_index == rhs.operand_index;
49 return lhs.operand_index < rhs.operand_index;
162 const std::function<void(Instruction*, uint32_t operand_index)>& f) const;
165 const std::function<void(Instruction*, uint32_t operand_index)>& f) const;
177 const std::function<bool(Instruction*, uint32_t operand_index)>& f) const;
180 const std::function<bool(Instruction*, uint32_t operand_index)>& f) const;
Dgraphics_robust_access_pass.cpp276 auto replace_index = [this, &inst, def_use_mgr](uint32_t operand_index, in ClampIndicesForAccessChain()
278 inst.SetOperand(operand_index, {new_value->result_id()}); in ClampIndicesForAccessChain()
291 uint32_t operand_index, Instruction* old_value, in ClampIndicesForAccessChain() argument
295 return replace_index(operand_index, clamp_inst); in ClampIndicesForAccessChain()
303 &clamp_index](uint32_t operand_index, uint64_t count) -> spv_result_t { in ClampIndicesForAccessChain() argument
305 this->GetDef(inst.GetSingleWordOperand(operand_index)); in ClampIndicesForAccessChain()
313 return replace_index(operand_index, GetValueForType(0, index_type)); in ClampIndicesForAccessChain()
340 << operand_index << " of access chain " in ClampIndicesForAccessChain()
361 return replace_index(operand_index, GetValueForType(0, index_type)); in ClampIndicesForAccessChain()
368 return replace_index(operand_index, in ClampIndicesForAccessChain()
[all …]
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Ddef_use_manager.h36 uint32_t operand_index; // logical operand index of the id use. This can be member
41 return lhs.inst == rhs.inst && lhs.operand_index == rhs.operand_index;
49 return lhs.operand_index < rhs.operand_index;
162 const std::function<void(Instruction*, uint32_t operand_index)>& f) const;
165 const std::function<void(Instruction*, uint32_t operand_index)>& f) const;
177 const std::function<bool(Instruction*, uint32_t operand_index)>& f) const;
180 const std::function<bool(Instruction*, uint32_t operand_index)>& f) const;
Dgraphics_robust_access_pass.cpp270 auto replace_index = [this, &inst, def_use_mgr](uint32_t operand_index, in ClampIndicesForAccessChain()
272 inst.SetOperand(operand_index, {new_value->result_id()}); in ClampIndicesForAccessChain()
285 uint32_t operand_index, Instruction* old_value, in ClampIndicesForAccessChain() argument
289 return replace_index(operand_index, clamp_inst); in ClampIndicesForAccessChain()
297 &clamp_index](uint32_t operand_index, uint64_t count) -> spv_result_t { in ClampIndicesForAccessChain() argument
299 this->GetDef(inst.GetSingleWordOperand(operand_index)); in ClampIndicesForAccessChain()
307 return replace_index(operand_index, GetValueForType(0, index_type)); in ClampIndicesForAccessChain()
334 << operand_index << " of access chain " in ClampIndicesForAccessChain()
355 return replace_index(operand_index, GetValueForType(0, index_type)); in ClampIndicesForAccessChain()
362 return replace_index(operand_index, in ClampIndicesForAccessChain()
[all …]
/external/deqp-deps/SPIRV-Tools/source/opt/
Ddef_use_manager.h36 uint32_t operand_index; // logical operand index of the id use. This can be member
41 return lhs.inst == rhs.inst && lhs.operand_index == rhs.operand_index;
49 return lhs.operand_index < rhs.operand_index;
162 const std::function<void(Instruction*, uint32_t operand_index)>& f) const;
165 const std::function<void(Instruction*, uint32_t operand_index)>& f) const;
177 const std::function<bool(Instruction*, uint32_t operand_index)>& f) const;
180 const std::function<bool(Instruction*, uint32_t operand_index)>& f) const;
Dspread_volatile_semantics.cpp140 for (uint32_t operand_index = kOpEntryPointInOperandInterface; in HasInterfaceInConflictOfVolatileSemantics() local
141 operand_index < entry_point.NumInOperands(); ++operand_index) { in HasInterfaceInConflictOfVolatileSemantics()
142 uint32_t var_id = entry_point.GetSingleWordInOperand(operand_index); in HasInterfaceInConflictOfVolatileSemantics()
175 for (uint32_t operand_index = kOpEntryPointInOperandInterface; in CollectTargetsForVolatileSemantics() local
176 operand_index < entry_point.NumInOperands(); ++operand_index) { in CollectTargetsForVolatileSemantics()
177 uint32_t var_id = entry_point.GetSingleWordInOperand(operand_index); in CollectTargetsForVolatileSemantics()
Dgraphics_robust_access_pass.cpp275 auto replace_index = [this, &inst, def_use_mgr](uint32_t operand_index, in ClampIndicesForAccessChain()
277 inst.SetOperand(operand_index, {new_value->result_id()}); in ClampIndicesForAccessChain()
290 uint32_t operand_index, Instruction* old_value, in ClampIndicesForAccessChain() argument
294 return replace_index(operand_index, clamp_inst); in ClampIndicesForAccessChain()
302 &clamp_index](uint32_t operand_index, uint64_t count) -> spv_result_t { in ClampIndicesForAccessChain() argument
304 this->GetDef(inst.GetSingleWordOperand(operand_index)); in ClampIndicesForAccessChain()
312 return replace_index(operand_index, GetValueForType(0, index_type)); in ClampIndicesForAccessChain()
339 << operand_index << " of access chain " in ClampIndicesForAccessChain()
360 return replace_index(operand_index, GetValueForType(0, index_type)); in ClampIndicesForAccessChain()
367 return replace_index(operand_index, in ClampIndicesForAccessChain()
[all …]
/external/deqp-deps/SPIRV-Tools/source/reduce/
Dchange_operand_reduction_opportunity.h31 uint32_t operand_index, uint32_t new_id) in ChangeOperandReductionOpportunity() argument
33 operand_index_(operand_index), in ChangeOperandReductionOpportunity()
34 original_id_(inst->GetOperand(operand_index).words[0]), in ChangeOperandReductionOpportunity()
35 original_type_(inst->GetOperand(operand_index).type), in ChangeOperandReductionOpportunity()
/external/swiftshader/third_party/SPIRV-Tools/source/reduce/
Dchange_operand_reduction_opportunity.h31 uint32_t operand_index, uint32_t new_id) in ChangeOperandReductionOpportunity() argument
33 operand_index_(operand_index), in ChangeOperandReductionOpportunity()
34 original_id_(inst->GetOperand(operand_index).words[0]), in ChangeOperandReductionOpportunity()
35 original_type_(inst->GetOperand(operand_index).type), in ChangeOperandReductionOpportunity()
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/reduce/
Dchange_operand_reduction_opportunity.h31 uint32_t operand_index, uint32_t new_id) in ChangeOperandReductionOpportunity() argument
33 operand_index_(operand_index), in ChangeOperandReductionOpportunity()
34 original_id_(inst->GetOperand(operand_index).words[0]), in ChangeOperandReductionOpportunity()
35 original_type_(inst->GetOperand(operand_index).type), in ChangeOperandReductionOpportunity()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion.cc72 int64_t operand_index) { in ShouldFuse() argument
73 HloInstruction* producer = consumer->mutable_operand(operand_index); in ShouldFuse()
74 VLOG(2) << "Considering for fusion: operand " << operand_index << " of " in ShouldFuse()
95 ReusesOperandElements(consumer, operand_index)) { in ShouldFuse()
100 !InstructionFusion::ShouldFuse(consumer, operand_index)) { in ShouldFuse()
150 if (consumer->operand(0)->shape().rank() == 1 && operand_index == 1 && in ShouldFuse()
156 operand_index == 0 && in ShouldFuse()
/external/deqp-deps/SPIRV-Tools/test/fuzz/
Dtransformation_permute_phi_operands_test.cpp117 [&entry](opt::Instruction* inst, uint32_t operand_index) -> bool { in TEST() argument
119 EXPECT_EQ(entry.second, operand_index); in TEST()
130 opt::Instruction* inst, uint32_t operand_index) { in TEST() argument
136 if (operand_index == 2) { in TEST()
140 ASSERT_EQ(3, operand_index); in TEST()
/external/angle/third_party/vulkan-deps/spirv-tools/src/test/fuzz/
Dtransformation_permute_phi_operands_test.cpp117 [&entry](opt::Instruction* inst, uint32_t operand_index) -> bool { in TEST() argument
119 EXPECT_EQ(entry.second, operand_index); in TEST()
130 opt::Instruction* inst, uint32_t operand_index) { in TEST() argument
136 if (operand_index == 2) { in TEST()
140 ASSERT_EQ(3, operand_index); in TEST()
/external/swiftshader/third_party/SPIRV-Tools/test/fuzz/
Dtransformation_permute_phi_operands_test.cpp117 [&entry](opt::Instruction* inst, uint32_t operand_index) -> bool { in TEST() argument
119 EXPECT_EQ(entry.second, operand_index); in TEST()
130 opt::Instruction* inst, uint32_t operand_index) { in TEST() argument
136 if (operand_index == 2) { in TEST()
140 ASSERT_EQ(3, operand_index); in TEST()

1234567