Home
last modified time | relevance | path

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

/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/lite/delegates/xnnpack/
Dreduce_tester.h110 bool keep_dims_ = true; variable
Dfully_connected_tester.h130 bool keep_dims_ = false; variable
Dquantized_fully_connected_tester.h176 bool keep_dims_ = false; variable
/external/tensorflow/tensorflow/core/kernels/
Dsparse_reduce_op.cc245 bool keep_dims_; member in tensorflow::SparseReduceOp
343 bool keep_dims_; member in tensorflow::SparseReduceSparseOp
Dreduce_join_op.cc180 bool keep_dims_; member in tensorflow::ReduceJoinOp