| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | comparators.cc | 38 const std::string& name, const std::vector<PrimitiveType>& operand_types, in CreateScalarComparisonComputation()
49 const std::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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/opdefs/ |
| H A D | tfrt_fallback_common.cc | 94 llvm::SmallVector<mlir::Type, 4> operand_types; in ParseExecuteOpCommon() local
|
| H A D | tfrt_fallback_async.cc | 173 SmallVector<Type, 4> operand_types; in parseBatchFunctionOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/ |
| H A D | tfrt_fallback_common.cc | 94 llvm::SmallVector<mlir::Type, 4> operand_types; in ParseExecuteOpCommon() local
|
| H A D | tfrt_fallback_async.cc | 239 SmallVector<Type, 4> operand_types; in parse() local
|
| /aosp_15_r20/external/vixl/tools/test_generator/ |
| H A D | parser.py | 54 def __init__(self, operand_types, operand_variants, input_types, argument
|
| H A D | generator.py | 149 def GenerateOperands(self, operand_types): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | mlir_emitter.cc | 104 std::vector<mlir::Type> operand_types = {ret_memref}; in EmitMlirFuncAndCall() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | cluster_outlining.cc | 60 llvm::SmallVector<Type, 4> operand_types; in BuildFunction() local
|
| H A D | tpu_extract_outside_compilation.cc | 70 llvm::SmallVector<Type, 4> operand_types; in BuildFunction() local
|
| H A D | shape_inference.cc | 2343 TypeRange operand_types, TypeRange result_types, in GetWhileCompatibleTypes()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| H A D | tf_jitrt_legalize_i1_type.cc | 218 auto operand_types = op->getOperandTypes(); in runOnOperation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | ops.cc | 388 std::vector<PrimitiveType> operand_types; in BuildOpsSubmodule() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| H A D | tf_to_jit_invocations.cc | 126 auto operand_types = llvm::to_vector<16>(llvm::map_range( in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | bfloat16_normalization.cc | 262 std::vector<PrimitiveType> operand_types(hlo->operand_count()); in HandleMultipleOutputs() local
|
| H A D | hlo_creation_utils.cc | 540 std::vector<PrimitiveType> operand_types(operands.size()); in MakeSortHlo() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf_communication.cc | 762 SmallVector<Type> operand_types; in ProcessRegionWhileOp() local
|
| H A D | legalize_tf.cc | 2246 SmallVector<Type, 3> operand_types = {act.getType(), feature_type, in matchAndRewrite() local
2380 SmallVector<Type, 3> operand_types = {bn_train_input_type_tensor, in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| H A D | tfl_ops.cc | 744 ArrayRef<TensorType> operand_types, in VerifyConcatenationOpTypes()
879 SmallVector<TensorType, 4> operand_types; in verify() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| H A D | tf_to_tfrt.cc | 2033 llvm::SmallVector<Type, 4> operand_types; in CreateCompiledModule() local
|