Home
last modified time | relevance | path

Searched defs:axis_ (Results 1 – 25 of 29) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantize_and_dequantize_op.cc137 int axis_; member in tensorflow::QuantizeAndDequantizeV2Op
229 int axis_; member in tensorflow::QuantizeAndDequantizeV4GradientOp
322 int axis_; member in tensorflow::QuantizeAndDequantizeV3Op
H A Dpack_op.cc119 int axis_; member in tensorflow::PackOp
H A Dunpack_op.cc121 int axis_; member in tensorflow::UnpackOp
H A Done_hot_op.cc124 int32 axis_; member in tensorflow::OneHotOp
H A Ddequantize_op.cc242 int axis_; member in tensorflow::DequantizeOp
H A Dquantize_op.cc344 int axis_; member in tensorflow::QuantizeV2Op
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dpack_op.cc86 int axis_; member in tensorflow::__anon848df4090111::PackOp
H A Done_hot_op.cc74 int32 axis_; member in tensorflow::__anona1ec7f760111::OneHotOp
H A Dunpack_op.cc85 int axis_; member in tensorflow::__anon9c13494c0111::UnpackOp
H A Dquantize_and_dequantize_op.cc202 int axis_; member in tensorflow::__anoncfa099260111::QuantizeAndDequantizeOp
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dconcatenation_tester.h86 int axis_; variable
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/
H A Dnormalize_op_test.py57 def _GetNormalizeOpTest(dtype_, shape_, ord_, axis_): argument
H A Dnorm_op_test.py63 def _GetNormOpTest(dtype_, shape_, ord_, axis_, keep_dims_, use_static_shape_): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dreduce_test_common.h66 int axis_; variable
H A Dexpand_dims_test.cc69 int axis_; member in tflite::__anonad5ed52a0111::ExpandDimsOpModel
H A Dcumsum_test.cc54 int axis_; member in tflite::ops::builtin::__anon301b62f80111::CumsumOpModel
H A Droll_test.cc88 int axis_; member in tflite::BaseRollOpModel
H A Dsplit_test.cc75 int axis_; member in tflite::__anon4a188e2a0111::SplitOpModel
H A Dreverse_test.cc51 int axis_; member in tflite::__anon717e067a0111::ReverseOpModel
H A Dsplit_v_test.cc86 int axis_; member in tflite::__anonc0c8999f0111::SplitVOpModel
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Dreduce_test.cc50 int axis_; member in tflite::ReduceOpModel
H A Dsplit_test.cc53 int axis_; member in tflite::SplitOpModel
/aosp_15_r20/external/pytorch/aten/src/ATen/quantized/
H A DQuantizer.h175 const int64_t axis_; member
/aosp_15_r20/external/executorch/backends/qualcomm/aot/wrappers/
H A DQuantizeParamsWrapper.h135 std::int32_t axis_; variable
/aosp_15_r20/external/rust/android-crates-io/crates/criterion-plot/src/
Daxis.rs163 let axis_ = axis.display(); in script() localVariable

12