Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Done_hot_op.cc74 int32 axis_; member in tensorflow::__anonc3bfa5090111::OneHotOp
Dpack_op.cc86 int axis_; member in tensorflow::__anon22a9ca7c0111::PackOp
Dunpack_op.cc85 int axis_; member in tensorflow::__anon307e887f0111::UnpackOp
/external/tensorflow/tensorflow/lite/kernels/
Dexpand_dims_test.cc48 int axis_; member in tflite::__anon459c72dd0111::ExpandDimsOpModel
Dsplit_test.cc64 int axis_; member in tflite::__anona5eaa19d0111::SplitOpModel
Dreverse_test.cc48 int axis_; member in tflite::__anon0a22c2ad0111::ReverseOpModel
Dsplit_v_test.cc71 int axis_; member in tflite::__anon596d55d20111::SplitVOpModel
Darg_min_max_test.cc44 int axis_; member in tflite::__anon0dc779d30111::ArgBaseOpModel
Dreduce_test.cc53 int axis_; member in tflite::__anonee8c12e90111::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::__anon6d004b400111::BaseReduceOpModel