Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpack_op.cc86 int axis_; member in tensorflow::__anone7c902130111::PackOp
Done_hot_op.cc74 int32 axis_; member in tensorflow::__anon86fa49400111::OneHotOp
Dunpack_op.cc85 int axis_; member in tensorflow::__anonba4a01d60111::UnpackOp
/external/tensorflow/tensorflow/lite/kernels/
Dexpand_dims_test.cc48 int axis_; member in tflite::__anon558ad8b40111::ExpandDimsOpModel
Dsplit_test.cc64 int axis_; member in tflite::__anon0e1332340111::SplitOpModel
Dreverse_test.cc48 int axis_; member in tflite::__anon1eb1d5040111::ReverseOpModel
Dsplit_v_test.cc71 int axis_; member in tflite::__anon6dfc68290111::SplitVOpModel
Darg_min_max_test.cc44 int axis_; member in tflite::__anon1db5dfaa0111::ArgBaseOpModel
Dreduce_test.cc53 int axis_; member in tflite::__anon5bc6b6600111::BaseOpModel
/external/tensorflow/tensorflow/python/kernel_tests/
Dnorm_op_test.py67 def _GetNormOpTest(dtype_, shape_, ord_, axis_, keep_dims_, use_static_shape_): argument
/external/tensorflow/tensorflow/core/kernels/
Dpack_op.cc132 int axis_; member in tensorflow::PackOp
Dunpack_op.cc126 int axis_; member in tensorflow::UnpackOp
Done_hot_op.cc123 int32 axis_; member in tensorflow::OneHotOp
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc3414 int axis_; member in tflite::__anon0c68fcb70111::BaseReduceOpModel