Searched defs:keep_dims_ (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ | ||
| D | norm_op_test.py | 67 def _GetNormOpTest(dtype_, shape_, ord_, axis_, keep_dims_, use_static_shape_): argument |
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ | ||
| D | fully_connected_tester.h | 111 bool keep_dims_ = false; variable |
| D | reduce_tester.h | 110 bool keep_dims_ = true; variable |
| /external/tensorflow/tensorflow/core/kernels/ | ||
| D | sparse_reduce_op.cc | 232 bool keep_dims_; member in tensorflow::SparseReduceOp |
| D | reduce_join_op.cc | 180 bool keep_dims_; member in tensorflow::ReduceJoinOp |