Searched defs:keep_dims_ (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/python/kernel_tests/linalg/ | ||
D | norm_op_test.py | 63 def _GetNormOpTest(dtype_, shape_, ord_, axis_, keep_dims_, use_static_shape_): argument |
/external/tensorflow/tensorflow/lite/delegates/xnnpack/ | ||
D | reduce_tester.h | 110 bool keep_dims_ = true; variable |
D | fully_connected_tester.h | 146 bool keep_dims_ = false; variable |
D | quantized_reduce_tester.h | 144 bool keep_dims_ = true; variable |
D | quantized_fully_connected_tester.h | 183 bool keep_dims_ = false; variable |
/external/tensorflow/tensorflow/core/kernels/ | ||
D | sparse_reduce_op.cc | 250 bool keep_dims_; member in tensorflow::SparseReduceOp |
D | reduce_join_op.cc | 181 bool keep_dims_; member in tensorflow::ReduceJoinOp |