Home
last modified time | relevance | path

Searched defs:keep_dims_ (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dnorm_op_test.py63 def _GetNormOpTest(dtype_, shape_, ord_, axis_, keep_dims_, use_static_shape_): argument
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dreduce_tester.h110 bool keep_dims_ = true; variable
Dfully_connected_tester.h146 bool keep_dims_ = false; variable
Dquantized_reduce_tester.h144 bool keep_dims_ = true; variable
Dquantized_fully_connected_tester.h183 bool keep_dims_ = false; variable
/external/tensorflow/tensorflow/core/kernels/
Dsparse_reduce_op.cc250 bool keep_dims_; member in tensorflow::SparseReduceOp
351 bool keep_dims_; member in tensorflow::SparseReduceSparseOp
Dreduce_join_op.cc181 bool keep_dims_; member in tensorflow::ReduceJoinOp