Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Dquantize_and_dequantize_op.cc135 int axis_; member in tensorflow::QuantizeAndDequantizeV2Op
225 int axis_; member in tensorflow::QuantizeAndDequantizeV4GradientOp
314 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.cc237 int axis_; member in tensorflow::DequantizeOp
Dquantize_op.cc344 int axis_; member in tensorflow::QuantizeV2Op
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpack_op.cc86 int axis_; member in tensorflow::__anon4622a2bf0111::PackOp
Done_hot_op.cc74 int32 axis_; member in tensorflow::__anon4ace3a6c0111::OneHotOp
Dunpack_op.cc85 int axis_; member in tensorflow::__anon158e7d820111::UnpackOp
Dquantize_and_dequantize_op.cc202 int axis_; member in tensorflow::__anon8615801c0111::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::__anon356c5e600111::ExpandDimsOpModel
Dcumsum_test.cc54 int axis_; member in tflite::ops::builtin::__anon1eead26e0111::CumsumOpModel
Droll_test.cc88 int axis_; member in tflite::BaseRollOpModel
Dsplit_v_test.cc75 int axis_; member in tflite::__anon8985f7d50111::SplitVOpModel
Dsplit_test.cc75 int axis_; member in tflite::__anon705ce6e00111::SplitOpModel
Dreverse_test.cc51 int axis_; member in tflite::__anon3a3b64b00111::ReverseOpModel
Darg_min_max_test.cc78 int axis_; member in tflite::__anonfd9765560111::ArgBaseOpModel
Dreduce_test.cc75 int axis_; member in tflite::__anon0747008c0111::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.cc490 int axis_; member in tensorflow::MklQuantizeV2Op
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_device_selection_test.cc262 int axis_; member in tflite::__anone5dffd370111::ArgMaxOpModel

12