Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_support.cc24 int64 operand_index) const { in SupportsBF16Operand()
35 CHECK_EQ(operand_index, 0); in SupportsBF16Operand()
79 const HloInstruction& hlo, int64 operand_index) { in EffectiveOperandPrecisionIsOutputPrecision() argument
106 return operand_index == 0; in EffectiveOperandPrecisionIsOutputPrecision()
108 return operand_index == 0 || operand_index == 1; in EffectiveOperandPrecisionIsOutputPrecision()
110 return operand_index == 0; in EffectiveOperandPrecisionIsOutputPrecision()
113 return operand_index == 1 || operand_index == 2; in EffectiveOperandPrecisionIsOutputPrecision()
136 const HloInstruction& hlo, int64 operand_index) const { in EffectiveOperandPrecisionIsBF16()
Ddynamic_dimension_inference.cc156 int64 operand_index, HloInstruction* dynamic_size)>;
162 int64 operand_index, int64 dimension,
166 int64 operand_index,
170 int64 operand_index,
175 int64 operand_index, int64 dimension);
180 int64 operand_index,
203 int64 operand_index, HloInstruction* dynamic_size) { in DefaultAction() argument
215 int64 operand_index, HloInstruction* dynamic_size) { in HandleGetTupleElement() argument
228 int64 operand_index, HloInstruction* dynamic_size) { in HandleTuple() argument
229 index.push_front(operand_index); in HandleTuple()
[all …]
Dbfloat16_support.h32 int64 operand_index) const;
50 const HloInstruction& hlo, int64 operand_index);
55 int64 operand_index) const;
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion.cc58 int64 operand_index) { in ShouldFuseInexpensiveChecks() argument
59 HloInstruction* producer = consumer->mutable_operand(operand_index); in ShouldFuseInexpensiveChecks()
69 ReusesOperandElements(consumer, operand_index)) { in ShouldFuseInexpensiveChecks()
76 !InstructionFusion::ShouldFuse(consumer, operand_index)) { in ShouldFuseInexpensiveChecks()
85 int64 operand_index) { in ShouldFuse() argument
86 if (!ShouldFuseInexpensiveChecks(consumer, operand_index)) { in ShouldFuse()
87 VLOG(5) << "Not fusing inexpensive checks of operand " << operand_index in ShouldFuse()
91 auto producer = consumer->operand(operand_index); in ShouldFuse()
124 int64 operand_index) { in ShouldFuseIntoMultiOutput() argument
/external/angle/third_party/vulkan-deps/spirv-tools/src/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.cpp86 for (size_t operand_index = 2; operand_index < inst->operands().size(); in BitwisePass() local
87 ++operand_index) { in BitwisePass()
88 const uint32_t type_id = _.GetOperandTypeId(inst, operand_index); in BitwisePass()
94 << operand_index; in BitwisePass()
100 << " operand index " << operand_index; in BitwisePass()
106 << " operand index " << operand_index; in BitwisePass()
/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.cpp86 for (size_t operand_index = 2; operand_index < inst->operands().size(); in BitwisePass() local
87 ++operand_index) { in BitwisePass()
88 const uint32_t type_id = _.GetOperandTypeId(inst, operand_index); in BitwisePass()
94 << operand_index; in BitwisePass()
100 << " operand index " << operand_index; in BitwisePass()
106 << " operand index " << operand_index; in BitwisePass()
/external/swiftshader/third_party/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.cpp86 for (size_t operand_index = 2; operand_index < inst->operands().size(); in BitwisePass() local
87 ++operand_index) { in BitwisePass()
88 const uint32_t type_id = _.GetOperandTypeId(inst, operand_index); in BitwisePass()
94 << operand_index; in BitwisePass()
100 << " operand index " << operand_index; in BitwisePass()
106 << " operand index " << operand_index; in BitwisePass()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Ddef_use_manager.h38 uint32_t operand_index; // logical operand index of the id use. This can be member
43 return lhs.inst == rhs.inst && lhs.operand_index == rhs.operand_index;
51 return lhs.operand_index < rhs.operand_index;
163 const std::function<void(Instruction*, uint32_t operand_index)>& f) const;
166 const std::function<void(Instruction*, uint32_t operand_index)>& f) const;
178 const std::function<bool(Instruction*, uint32_t operand_index)>& f) const;
181 const std::function<bool(Instruction*, uint32_t operand_index)>& f) const;
Dgraphics_robust_access_pass.cpp275 auto replace_index = [&inst, def_use_mgr](uint32_t operand_index, in ClampIndicesForAccessChain()
277 inst.SetOperand(operand_index, {new_value->result_id()}); in ClampIndicesForAccessChain()
289 uint32_t operand_index, Instruction* old_value, in ClampIndicesForAccessChain() argument
293 return replace_index(operand_index, clamp_inst); in ClampIndicesForAccessChain()
301 &clamp_index](uint32_t operand_index, uint64_t count) -> spv_result_t { in ClampIndicesForAccessChain() argument
303 this->GetDef(inst.GetSingleWordOperand(operand_index)); in ClampIndicesForAccessChain()
311 return replace_index(operand_index, GetValueForType(0, index_type)); in ClampIndicesForAccessChain()
338 << operand_index << " of access chain " in ClampIndicesForAccessChain()
359 return replace_index(operand_index, GetValueForType(0, index_type)); in ClampIndicesForAccessChain()
366 return replace_index(operand_index, in ClampIndicesForAccessChain()
[all …]
/external/deqp-deps/SPIRV-Tools/source/opt/
Ddef_use_manager.h38 uint32_t operand_index; // logical operand index of the id use. This can be member
43 return lhs.inst == rhs.inst && lhs.operand_index == rhs.operand_index;
51 return lhs.operand_index < rhs.operand_index;
163 const std::function<void(Instruction*, uint32_t operand_index)>& f) const;
166 const std::function<void(Instruction*, uint32_t operand_index)>& f) const;
178 const std::function<bool(Instruction*, uint32_t operand_index)>& f) const;
181 const std::function<bool(Instruction*, uint32_t operand_index)>& f) const;
Dgraphics_robust_access_pass.cpp275 auto replace_index = [&inst, def_use_mgr](uint32_t operand_index, in ClampIndicesForAccessChain()
277 inst.SetOperand(operand_index, {new_value->result_id()}); in ClampIndicesForAccessChain()
289 uint32_t operand_index, Instruction* old_value, in ClampIndicesForAccessChain() argument
293 return replace_index(operand_index, clamp_inst); in ClampIndicesForAccessChain()
301 &clamp_index](uint32_t operand_index, uint64_t count) -> spv_result_t { in ClampIndicesForAccessChain() argument
303 this->GetDef(inst.GetSingleWordOperand(operand_index)); in ClampIndicesForAccessChain()
311 return replace_index(operand_index, GetValueForType(0, index_type)); in ClampIndicesForAccessChain()
338 << operand_index << " of access chain " in ClampIndicesForAccessChain()
359 return replace_index(operand_index, GetValueForType(0, index_type)); in ClampIndicesForAccessChain()
366 return replace_index(operand_index, in ClampIndicesForAccessChain()
[all …]
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Ddef_use_manager.h38 uint32_t operand_index; // logical operand index of the id use. This can be member
43 return lhs.inst == rhs.inst && lhs.operand_index == rhs.operand_index;
51 return lhs.operand_index < rhs.operand_index;
163 const std::function<void(Instruction*, uint32_t operand_index)>& f) const;
166 const std::function<void(Instruction*, uint32_t operand_index)>& f) const;
178 const std::function<bool(Instruction*, uint32_t operand_index)>& f) const;
181 const std::function<bool(Instruction*, uint32_t operand_index)>& f) const;
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/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()
Dchange_operand_to_undef_reduction_opportunity.h32 uint32_t operand_index) in ChangeOperandToUndefReductionOpportunity() argument
35 operand_index_(operand_index), in ChangeOperandToUndefReductionOpportunity()
36 original_id_(inst->GetOperand(operand_index).words[0]) {} in ChangeOperandToUndefReductionOpportunity()
/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/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion.cc72 int64 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()
98 ReusesOperandElements(consumer, operand_index)) { in ShouldFuse()
103 if (!InstructionFusion::ShouldFuse(consumer, operand_index)) { in ShouldFuse()
156 if (consumer->operand(0)->shape().rank() == 1 && operand_index == 1 && in ShouldFuse()
162 operand_index == 0 && in ShouldFuse()
/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()

123456