Searched defs:operand_types (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | comparators.cc | 38 const string& name, const std::vector<PrimitiveType>& operand_types, in CreateScalarComparisonComputation()
49 const string& name, const std::vector<PrimitiveType>& operand_types, in CreateScalarComparisonComputation()
115 const std::vector<PrimitiveType>& operand_types, XlaBuilder* builder) { in CreateScalarLtComputation()
122 const std::vector<PrimitiveType>& operand_types, XlaBuilder* builder) { in CreateScalarGtComputation()
|
| /external/llvm-project/mlir/lib/Dialect/LLVMIR/IR/ |
| D | NVVMDialect.cpp | 97 SmallVector<Type, 12> operand_types(op.getOperandTypes().begin(), in verify() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | cluster_outlining.cc | 55 llvm::SmallVector<Type, 4> operand_types; in BuildFunction() local
|
| D | shape_inference.cc | 1293 TypeRange operand_types, TypeRange result_types, in GetWhileCompatibleTypes()
|
| /external/vixl/tools/test_generator/ |
| D | parser.py | 54 def __init__(self, operand_types, operand_variants, input_types, argument
|
| D | generator.py | 149 def GenerateOperands(self, operand_types): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | mlir_emitter.cc | 101 std::vector<mlir::Type> operand_types = {ret_memref}; in EmitMlirFuncAndCall() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 272 std::vector<PrimitiveType> operand_types; in BuildOpsSubmodule() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/ |
| D | passes.cc | 255 auto operand_types = llvm::to_vector<4>(llvm::concat<const mlir::Type>( in runOnFunction() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_normalization.cc | 263 std::vector<PrimitiveType> operand_types(hlo->operand_count()); in HandleMultipleOutputs() local
|
| D | hlo_creation_utils.cc | 423 std::vector<PrimitiveType> operand_types(operands.size()); in MakeSortHlo() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 595 ArrayRef<TensorType> operand_types, in VerifyConcatenationOpTypes()
678 SmallVector<TensorType, 4> operand_types; in Verify() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1975 SmallVector<Type, 3> operand_types = {bn_train_input_type_tensor, in matchAndRewrite() local
|