Home
last modified time | relevance | path

Searched defs:input_type (Results 1 – 25 of 273) sorted by relevance

1234567891011

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mlir_generated/
H A Dbase_gpu_op.h26 #define REGISTER_ALIASED_GPU_KERNEL(tf_op, mlir_op, input_type, output_type) \ argument
31 #define REGISTER_GPU_KERNEL(tf_op, input_type, output_type) \ argument
35 #define REGISTER_COMPLEX_GPU_KERNEL(tf_op, input_type, output_type) \ argument
38 #define REGISTER_GPU_KERNEL_NO_TYPE_CONSTRAINT(tf_op, input_type) \ argument
43 #define GENERATE_AND_REGISTER_UNARY_GPU_KERNEL(tf_op, input_type) \ argument
47 #define GENERATE_AND_REGISTER_UNARY_GPU_KERNEL2(tf_op, input_type, \ argument
53 tf_op, input_type, output_type, casted_input_type, casted_output_type) \ argument
58 #define GENERATE_AND_REGISTER_UNARY_JIT_GPU_KERNEL(tf_op, input_type) \ argument
62 #define GENERATE_UNARY_GPU_KERNEL(tf_op, input_type) \ argument
65 #define GENERATE_UNARY_GPU_KERNEL2(tf_op, input_type, output_type) \ argument
[all …]
H A Dbase_op.h140 #define MLIR_FUNCTION(tf_op, platform, input_type, output_type) \ argument
143 #define MLIR_OP(tf_op, platform, input_type, output_type) \ argument
148 #define REGISTER_ALIASED_KERNEL(tf_op, mlir_op, platform, input_type, \ argument
157 #define REGISTER_KERNEL(tf_op, platform, input_type, output_type, \ argument
162 #define REGISTER_COMPLEX_KERNEL(tf_op, platform, input_type, output_type) \ argument
170 #define REGISTER_KERNEL_NO_TYPE_CONSTRAINT(tf_op, platform, input_type) \ argument
176 #define GENERATE_AND_REGISTER_UNARY_KERNEL(tf_op, platform, input_type, \ argument
181 #define GENERATE_AND_REGISTER_UNARY_KERNEL2(tf_op, platform, input_type, \ argument
187 tf_op, platform, input_type, output_type, casted_input_type, \ argument
194 #define GENERATE_AND_REGISTER_UNARY_JIT_KERNEL(tf_op, platform, input_type, \ argument
[all …]
/aosp_15_r20/external/angle/third_party/spirv-tools/src/source/val/
H A Dvalidate_conversion.cpp41 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
71 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
103 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
134 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
169 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
205 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
241 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
255 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
296 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
316 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
[all …]
/aosp_15_r20/external/deqp-deps/SPIRV-Tools/source/val/
Dvalidate_conversion.cpp41 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
71 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
103 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
134 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
169 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
205 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
241 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
255 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
296 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
316 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
[all …]
/aosp_15_r20/external/swiftshader/third_party/SPIRV-Tools/source/val/
H A Dvalidate_conversion.cpp41 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
71 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
103 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
134 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
169 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
205 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
241 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
255 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
296 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
316 const uint32_t input_type = _.GetOperandTypeId(inst, 2); in ConversionPass() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dinput_lib_test.py65 input_type, argument
104 input_type, argument
161 input_type, argument
274 def _create_dataset_or_input_fn(self, input_type, input_fn): argument
321 def testOneDeviceCPU(self, input_type, api_type, iteration_type, distribution, argument
344 def testOneDeviceCPUMultiWorker(self, input_type, api_type, iteration_type, argument
370 def testTwoDevicesOneGPUOneCPU(self, input_type, api_type, iteration_type, argument
394 def testTPU(self, input_type, api_type, iteration_type, distribution, argument
425 def testTupleDataset(self, input_type, api_type, iteration_type, distribution, argument
460 def testTupleDatasetMultiworker(self, input_type, api_type, iteration_type, argument
[all …]
H A Dinput_lib_type_spec_test.py56 def testTypeSpec(self, input_type, distribution, argument
88 def testTypeSpecRoundTrip(self, input_type, argument
124 def testDoesNotTriggerFunctionTracing(self, input_type, distribution, argument
231 def testMostSpecificCompatibleType(self, input_type, distribution, argument
/aosp_15_r20/external/autotest/client/cros/input_playback/
H A Dinput_playback.py18 def __init__(self, input_type): argument
110 def has(self, input_type): argument
124 def emulate(self, input_type='mouse', property_file=None): argument
340 def _find_device_ids(self, node_dir, device_dir, input_type, name): argument
472 def playback(self, filepath, input_type='touchpad'): argument
492 def blocking_playback(self, filepath, input_type='touchpad'): argument
520 def blocking_playback_of_default_file(self, filename, input_type='mouse'): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc264 const TfLiteType input_type = context->tensors[input_id].type; in NeedInt8Conversion() local
284 const TfLiteType input_type = context->tensors[input_id].type; in NeedInt8Conversion() local
392 const TfLiteType input_type = context->tensors[input_id].type; in IsHybridOperator() local
400 const TfLiteType input_type = context->tensors[input_id].type; in IsHybridOperator() local
409 const TfLiteType input_type = context->tensors[input_id].type; in IsHybridOperator() local
417 const TfLiteType input_type = context->tensors[input_id].type; in IsHybridOperator() local
424 const TfLiteType input_type = context->tensors[input_id].type; in IsHybridOperator() local
2103 const auto input_type = context->tensors[node->inputs->data[0]].type; in ExpectIsFloatOperator() local
2112 const auto input_type = context->tensors[node->inputs->data[0]].type; in ExpectIsFloatOrUint8Operator() local
2121 const auto input_type = context->tensors[node->inputs->data[0]].type; in ExpectIsFloatOrQuant8Operator() local
[all …]
/aosp_15_r20/external/perfetto/src/trace_processor/perfetto_sql/stdlib/chrome/
H A Dinput.sql23 input_type STRING, field
34 EXTRACT_ARG(thread_slice.arg_set_id, 'chrome_latency_info.input_type') AS input_type, field
48 input_type STRING field
54 MIN(input_type) as input_type field
76 input_type STRING, field
87 chrome_inputs.input_type AS input_type, field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Ddevice_transform_patterns.cc154 RankedTensorType input_type, int split_dim, in SliceOutputs()
210 auto input_type = pack_inputs[0].getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
291 auto input_type = input.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
331 auto input_type = input.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local
372 auto input_type = input.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local
473 auto input_type = input.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local
533 auto input_type = input.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local
548 auto input_type = input.getType().cast<RankedTensorType>(); in matchAndRewrite() local
604 auto input_type = input.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local
669 auto input_type = input.getType().dyn_cast_or_null<ShapedType>(); in matchAndRewrite() local
H A Ddevice_transform.cc70 auto input_type = input.get().getType(); in ConvertQuantizedOpToFloat() local
86 auto input_type = input.get().getType(); in ConvertQuantizedOpToFloat() local
162 auto input_type = input_dequant.getType().dyn_cast<ShapedType>(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_common.cc145 RankedTensorType input_type = in convertPackOp() local
313 RankedTensorType input_type = in convertUnpackOp() local
458 RankedTensorType input_type = input.getType().dyn_cast<RankedTensorType>(); in convertZerosLikeOp() local
577 ShapedType input_type = input.getType().dyn_cast<ShapedType>(); in convertRoundOp() local
729 RankedTensorType input_type = in convertSpaceToBatchNDOp() local
992 RankedTensorType input_type = in convertBatchToSpaceNDOp() local
1193 RankedTensorType input_type = in convertExpandDimsOp() local
1256 RankedTensorType input_type = in convertSqueezeOp() local
1370 RankedTensorType input_type = in convertSoftmaxOp() local
1789 RankedTensorType input_type = in convertLogSoftmaxOp() local
[all …]
H A Dlegalize_tfl.cc181 ShapedType input_type = tfl_relu_op.x().getType().dyn_cast<ShapedType>(); in matchAndRewrite() local
231 ShapedType input_type = tfl_relu1_op.x().getType().dyn_cast<ShapedType>(); in matchAndRewrite() local
286 ShapedType input_type = tfl_relu6_op.x().getType().dyn_cast<ShapedType>(); in matchAndRewrite() local
644 ShapedType input_type = tfl_round_op.x().getType().dyn_cast<ShapedType>(); in matchAndRewrite() local
871 ShapedType input_type = in matchAndRewrite() local
948 ShapedType input_type = in matchAndRewrite() local
1003 RankedTensorType input_type = in matchAndRewrite() local
1096 ShapedType input_type = tfl_conv_op.input().getType().dyn_cast<ShapedType>(); in matchAndRewrite() local
1218 ShapedType input_type = in matchAndRewrite() local
1467 RankedTensorType input_type = in matchAndRewrite() local
[all …]
H A Dlegalize_utils.cc59 auto input_type = input_val.getType().dyn_cast<mlir::ShapedType>(); in buildRescaleToInt32() local
72 auto input_type = input_val.getType().dyn_cast<mlir::ShapedType>(); in buildRescaleFromInt32() local
84 Value conv_val, ShapedType input_type, in buildRescaleOpConvOutput()
357 ShapedType input_type, ShapedType filter_type, in getPaddingValuesFromPadType()
430 uint32_t first_filter_spatial_dim, ShapedType input_type, in getTransposeConv2dPaddingValues()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddebug_ops_test.cc43 Status Init(DataType input_type, const std::vector<string>& debug_urls) { in Init()
54 Status Init(DataType input_type) { in Init()
181 Status Init(DataType input_type) { in Init()
244 Status Init(DataType input_type) { in Init()
252 Status InitGated(DataType input_type, const std::vector<string>& debug_urls) { in InitGated()
635 Status Init(DataType input_type) { in Init()
687 Status Init(DataType input_type) { in Init()
/aosp_15_r20/external/perfetto/src/ipc/protoc_plugin/
H A Dipc_plugin.cc131 std::string input_type = method.input_type()->name(); in ForEachMethod() local
193 const std::string& input_type, in GenerateServiceHeader()
194 const std::string& output_type) { in GenerateServiceHeader()
237 const std::string& input_type, in GenerateServiceCpp()
238 const std::string& output_type) { in GenerateServiceCpp()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc990 auto input_type = op.input().getType().cast<ShapedType>(); in matchAndRewrite() local
1021 ShapedType input_type = op.input().getType().cast<ShapedType>(); in verify() local
1087 const auto input_type = input_tensor.getType().cast<ShapedType>(); in fold() local
1572 auto input_type = operand0.getType().cast<ShapedType>(); in verify() local
1602 auto input_type = op.input().getType().cast<ShapedType>(); in verify() local
1708 bool InputOutputHasSameShape(mlir::Type input_type, mlir::Type output_type) { in InputOutputHasSameShape()
1723 auto input_type = getOperand(0).getType().cast<ShapedType>(); in fold() local
2010 auto input_type = op.input().getType().cast<ShapedType>(); in verify() local
2264 auto input_type = operands[0].getType().dyn_cast<ShapedType>(); in inferReturnTypes() local
2336 RankedTensorType input_type, int64_t dim, int64_t dim_size) { in SubstituteRankedTensorTypeDimSize()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Darg_min_max_test.cc26 explicit ArgBaseOpModel(TensorType input_type) { in ArgBaseOpModel()
47 ArgMinOpModel(std::initializer_list<int> input_shape, TensorType input_type) in ArgMinOpModel()
63 ArgMaxOpModel(std::initializer_list<int> input_shape, TensorType input_type) in ArgMaxOpModel()
/aosp_15_r20/external/rust/android-crates-io/crates/grpcio-sys/grpc/src/compiler/
Dphp_generator.cc69 const Descriptor* input_type = method->input_type(); in PrintMethod() local
140 const Descriptor* input_type = method->input_type(); in PrintServerMethod() local
248 auto input_type = method->input_type(); in PrintServerMethodDescriptors() local
/aosp_15_r20/external/grpc-grpc/src/compiler/
H A Dphp_generator.cc69 const Descriptor* input_type = method->input_type(); in PrintMethod() local
140 const Descriptor* input_type = method->input_type(); in PrintServerMethod() local
248 auto input_type = method->input_type(); in PrintServerMethodDescriptors() local
/aosp_15_r20/external/autotest/client/cros/
H A Dtouch_playback_test_base.py77 def _find_test_files(self, input_type, gestures): argument
122 def _find_test_files_from_directions(self, input_type, fmt_str, directions): argument
144 def _download_remote_test_file(self, filename, input_type): argument
300 def _get_kernel_events_recorder(self, input_type): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dmodify_model_interface.cc51 std::vector<TensorOpTensor> GetInputTensors(const TensorType& input_type, in GetInputTensors()
302 int GetOriginalNumberOfTensors(const TensorType& input_type, in GetOriginalNumberOfTensors()
315 ModelT* model, const TensorType& input_type, in ModifyModelInterface()
363 const TensorType& input_type, in ModifyModelInterface()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dtile_test.cc60 TensorType input_type, TensorType multiply_type, in TileOpConstModel()
74 TensorType input_type, TensorType multiply_type) { in TileOpDynamicModel()
94 TensorType input_type, TensorType multiply_type, in Check()
H A Drandom_ops_test.cc34 RandomOpModel(BuiltinOperator op_code, InputType input_type, in RandomOpModel()
65 MultinomialOpModel(InputType input_type, in MultinomialOpModel()
108 InputType input_type = std::get<1>(GetParam()); in TEST_P() local
137 InputType input_type = std::get<1>(GetParam()); in TEST_P() local

1234567891011