Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dquantize_and_dequantize_op.cc131 int axis_; member in tensorflow::QuantizeAndDequantizeV2Op
204 int axis_; member in tensorflow::QuantizeAndDequantizeV4GradientOp
289 int axis_; member in tensorflow::QuantizeAndDequantizeV3Op
Dpack_op.cc120 int axis_; member in tensorflow::PackOp
Dunpack_op.cc122 int axis_; member in tensorflow::UnpackOp
Done_hot_op.cc124 int32 axis_; member in tensorflow::OneHotOp
Ddequantize_op.cc225 int axis_; member in tensorflow::DequantizeOp
Dquantize_op.cc301 int axis_; member in tensorflow::QuantizeV2Op
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Done_hot_op.cc74 int32 axis_; member in tensorflow::__anon86fa49400111::OneHotOp
Dpack_op.cc86 int axis_; member in tensorflow::__anone7c902130111::PackOp
Dunpack_op.cc85 int axis_; member in tensorflow::__anonba4a01d60111::UnpackOp
Dquantize_and_dequantize_op.cc202 int axis_; member in tensorflow::__anon210cb6f00111::QuantizeAndDequantizeOp
/external/tensorflow/tensorflow/python/kernel_tests/
Dnormalize_op_test.py61 def _GetNormalizeOpTest(dtype_, shape_, ord_, axis_): argument
Dnorm_op_test.py67 def _GetNormOpTest(dtype_, shape_, ord_, axis_, keep_dims_, use_static_shape_): argument
/external/tensorflow/tensorflow/lite/kernels/
Dexpand_dims_test.cc69 int axis_; member in tflite::__anon558ad8b40111::ExpandDimsOpModel
Dcumsum_test.cc54 int axis_; member in tflite::ops::builtin::__anon736a88420111::CumsumOpModel
Dsplit_v_test.cc75 int axis_; member in tflite::__anon6dfc68290111::SplitVOpModel
Dsplit_test.cc75 int axis_; member in tflite::__anon0e1332340111::SplitOpModel
Dreverse_test.cc51 int axis_; member in tflite::__anon1eb1d5040111::ReverseOpModel
Darg_min_max_test.cc78 int axis_; member in tflite::__anon1db5dfaa0111::ArgBaseOpModel
Dreduce_test.cc75 int axis_; member in tflite::__anon5bc6b6600111::BaseOpModel
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dreduce_test.cc48 int axis_; member in tflite::ReduceOpModel
Dsplit_test.cc50 int axis_; member in tflite::SplitOpModel
/external/rust/crates/criterion-plot/src/
Daxis.rs160 let axis_ = axis.display(); in script() localVariable
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_quantize_op.cc489 int axis_; member in tensorflow::MklQuantizeV2Op
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_device_selection_test.cc262 int axis_; member in tflite::__anon2e62bd8b0111::ArgMaxOpModel
Dnnapi_delegate_test.cc4677 int axis_; member in tflite::__anon0c68fcb70111::BaseReduceOpModel