Searched refs:operand_types (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | comparators.cc | 93 const string& name, const std::vector<PrimitiveType>& operand_types, in CreateScalarComparisonComputation() argument
98 if (operand_types.empty()) { in CreateScalarComparisonComputation()
112 for (auto operand_type : operand_types) { in CreateScalarComparisonComputation()
124 if (primitive_util::IsFloatingPointType(operand_types[0])) { in CreateScalarComparisonComputation()
125 PrimitiveType compare_type = operand_types[0]; in CreateScalarComparisonComputation()
147 const std::vector<PrimitiveType>& operand_types, XlaBuilder* builder) { in CreateScalarLtComputation() argument
148 return CreateScalarComparisonComputation("compare-less-than", operand_types, in CreateScalarLtComputation()
154 const std::vector<PrimitiveType>& operand_types, XlaBuilder* builder) { in CreateScalarGtComputation() argument
156 operand_types, builder, Gt); in CreateScalarGtComputation()
|
| D | comparators.h | 34 const std::vector<PrimitiveType>& operand_types, XlaBuilder* builder);
43 const std::vector<PrimitiveType>& operand_types, XlaBuilder* builder);
|
| /external/vixl/tools/test_generator/ |
| D | parser.py | 54 def __init__(self, operand_types, operand_variants, input_types, argument
56 self.operand_types = operand_types
66 type_name = self.operand_types[identifier]
142 operand_types = {
160 return DataTypeBuilder(operand_types, operand_variants, input_types, input_values)
|
| D | generator.py | 149 def GenerateOperands(self, operand_types): argument
180 for operand_type in operand_types.unwrap()
183 args=",".join(operand_types.GetNames()),
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_normalization.cc | 154 std::vector<PrimitiveType> operand_types(hlo->operand_count()); in HandleMultipleOutputs() local
161 operand_types[i] = hlo->operand(i)->shape().element_type(); in HandleMultipleOutputs()
163 if (operand_types[i] == F32) { in HandleMultipleOutputs()
165 } else if (operand_types[i] == BF16) { in HandleMultipleOutputs()
186 if (operand_types[i] != BF16) { in HandleMultipleOutputs()
|
| D | hlo_creation_utils.cc | 295 std::vector<PrimitiveType> operand_types(operands.size()); in MakeSortHlo() local
297 operand_types[i] = operands[i]->shape().element_type(); in MakeSortHlo()
299 XlaComputation comparator = CreateScalarLtComputation(operand_types, &b); in MakeSortHlo()
|
| /external/v8/src/compiler/ |
| D | bytecode-analysis.cc | 98 const OperandType* operand_types = Bytecodes::GetOperandTypes(bytecode); in UpdateInLiveness() local
119 switch (operand_types[i]) { in UpdateInLiveness()
159 DCHECK(!Bytecodes::IsRegisterOutputOperandType(operand_types[i])); in UpdateInLiveness()
168 switch (operand_types[i]) { in UpdateInLiveness()
197 DCHECK(!Bytecodes::IsRegisterInputOperandType(operand_types[i])); in UpdateInLiveness()
278 const OperandType* operand_types = Bytecodes::GetOperandTypes(bytecode); in UpdateAssignments() local
281 switch (operand_types[i]) { in UpdateAssignments()
301 DCHECK(!Bytecodes::IsRegisterOutputOperandType(operand_types[i])); in UpdateAssignments()
|
| /external/v8/src/interpreter/ |
| D | bytecode-array-accessor.cc | 161 const OperandType* operand_types = in GetRegisterOperandRange() local
163 OperandType operand_type = operand_types[operand_index]; in GetRegisterOperandRange()
|
| D | bytecode-array-builder.cc | 293 OperandType... operand_types>
309 return BytecodeNode::Create<bytecode, accumulator_use, operand_types...>( in Make()
311 OperandHelper<operand_types>::Convert(builder, operands)...); in Make()
|
| D | bytecode-node.h | 131 OperandType... operand_types>
|
| D | bytecode-array-builder.h | 534 OperandType... operand_types>
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1636 std::vector<PrimitiveType> operand_types; in Sort() local
1639 operand_types.push_back(operand_shape.element_type()); in Sort()
1646 for (auto operand_type : operand_types) { in Sort()
1660 if (primitive_util::IsFloatingPointType(operand_types[0])) { in Sort()
1661 PrimitiveType compare_type = operand_types[0]; in Sort()
|