Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dsparse_reduce_op.cc157 OP_REQUIRES_OK(ctx, ctx->GetAttr("keep_dims", &keep_dims_)); in SparseReduceOp()
180 sp, reduction_axes_t->flat<int32>(), keep_dims_); in Compute()
232 bool keep_dims_; member in tensorflow::SparseReduceOp
253 OP_REQUIRES_OK(ctx, ctx->GetAttr("keep_dims", &keep_dims_)); in SparseReduceSparseOp()
270 sp, reduction_axes_t->flat<int32>(), keep_dims_); in Compute()
304 if (keep_dims_) { in Compute()
328 bool keep_dims_; member in tensorflow::SparseReduceSparseOp
Dreduce_join_op.cc119 OP_REQUIRES_OK(ctx, ctx->GetAttr("keep_dims", &keep_dims_)); in ReduceJoinOp()
156 GetOutputShape(index_is_reduced, input_shape, keep_dims_); in Compute()
181 bool keep_dims_; member in tensorflow::ReduceJoinOp
Dreduction_ops_common.h145 OP_REQUIRES_OK(ctx, ctx->GetAttr("keep_dims", &keep_dims_));
155 OP_REQUIRES_OK(ctx, helper.Simplify(data, axes, keep_dims_));
247 bool keep_dims_;
/external/tensorflow/tensorflow/python/kernel_tests/
Dnorm_op_test.py67 def _GetNormOpTest(dtype_, shape_, ord_, axis_, keep_dims_, use_static_shape_): argument
70 np_norm = np.linalg.norm(matrix, ord=ord_, axis=axis_, keepdims=keep_dims_)
75 tf_matrix, ord=ord_, axis=axis_, keepdims=keep_dims_)
80 tf_matrix, ord=ord_, axis=axis_, keepdims=keep_dims_)
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreduction_ops_common.cc33 OP_REQUIRES_OK(ctx, ctx->GetAttr("keep_dims", &keep_dims_)); in XlaReductionOp()
93 } else if (keep_dims_) { in Compile()
121 auto result = keep_dims_ ? xla::Reshape(finalized, final_shape) : finalized; in Compile()
Dreduction_ops.h64 bool keep_dims_;