| /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/tensorflow/tensorflow/core/runtime_fallback/opdefs/ |
| D | tfrt_fallback_common.cc | 94 llvm::SmallVector<mlir::Type, 4> operand_types; in ParseExecuteOpCommon() local
|
| D | tfrt_fallback_async.cc | 173 SmallVector<Type, 4> operand_types; in parseBatchFunctionOp() local
|
| /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/mlir/tensorflow/transforms/ |
| D | cluster_outlining.cc | 60 llvm::SmallVector<Type, 4> operand_types; in BuildFunction() local
|
| D | tpu_extract_outside_compilation.cc | 70 llvm::SmallVector<Type, 4> operand_types; in BuildFunction() local
|
| D | shape_inference.cc | 1520 TypeRange operand_types, TypeRange result_types, in GetWhileCompatibleTypes()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 343 std::vector<PrimitiveType> operand_types; in BuildOpsSubmodule() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops_common.cc | 56 LogicalResult VerifyReduceScatter(Operation *op, TypeRange operand_types, in VerifyReduceScatter()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_to_jit_invocations.cc | 100 auto operand_types = llvm::to_vector<16>( in matchAndRewrite() 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 | 425 std::vector<PrimitiveType> operand_types(operands.size()); in MakeSortHlo() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | rank_specialization.cc | 132 auto operand_types = llvm::to_vector<16>( in matchAndRewrite() local
216 auto operand_types = llvm::to_vector<16>( in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 655 ArrayRef<TensorType> operand_types, in VerifyConcatenationOpTypes()
738 SmallVector<TensorType, 4> operand_types; in Verify() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | tf_to_tfrt.cc | 1919 llvm::SmallVector<Type, 4> operand_types; in CreateCompiledModule() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2588 SmallVector<Type, 3> operand_types = {bn_train_input_type_tensor, in matchAndRewrite() local
|