Home
last modified time | relevance | path

Searched defs:float_op (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dregister_ops_utils.h462 #define DEFINE_GENERIC_OP(aten_op, int_op, float_op, int_result, float_result) \ argument
528 aten_op, int_op, float_op, result, string_val) \ argument
559 #define DEFINE_SCALAR_BINARY_OP(aten_op, int_op, float_op, result) \ argument
564 aten_op, int_op, float_op, result) \ argument
694 float_op, \ argument
771 aten_op, int_op, float_op, complex_op, result, string_val) \ argument
819 aten_op, int_op, float_op, complex_op, result) \ argument
860 aten_op, int_op, float_op, complex_op, result) \ argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
H A Dquantize_model_test.cc476 const auto float_op = in TEST_P() local
583 const auto float_op = in TEST_P() local
781 auto float_op = float_graph->operators()->Get(1); in TEST_F() local
830 auto float_op = float_graph->operators()->Get(0); in TEST_F() local
948 auto float_op = float_graph->operators()->Get(0); in TEST_F() local
1096 auto float_op = float_graph->operators()->Get(0); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
H A Dquantization_utils.h430 Operation* float_op = rewriter.clone(*quantizing_op); in matchAndRewrite() local
608 Operation* float_op) const { in RewireFloatModelBackbone()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Ddepthwise_conv_test.cc729 DepthwiseConvolutionOpModel float_op(GetRegistration(), in TEST_P() local
767 DepthwiseConvolutionOpModel float_op(GetRegistration(), in TEST_P() local
H A Dconv_test.cc1021 ConvolutionOpModel float_op( in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantize_model_test.cc239 const auto float_op = float_graph->operators()->Get(i); in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate_test.cc849 ConvolutionOpModel float_op({TensorType_FLOAT32, {2, 2, 4, 1}}, in TEST() local
1233 DepthwiseConvolutionOpModel float_op({TensorType_FLOAT32, {1, 3, 2, 2}}, in TEST() local
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py7588 self, float_module, float_op, quantized_module, quantized_op): argument
7659 self, float_module, float_op, op_args, data, quantized_module, quantized_op, argument
7694 self, float_module, float_op, op_args, data, argument
/aosp_15_r20/external/vixl/test/aarch64/
H A Dtest-simulator-aarch64.cc403 bool float_op = reg_size == kSRegSize; in Test2Op_Helper() local