Searched refs:keep_dims_ (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | sparse_reduce_op.cc | 157 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
|
D | reduce_join_op.cc | 119 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
|
D | reduction_ops_common.h | 145 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/ |
D | norm_op_test.py | 67 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/ |
D | reduction_ops_common.cc | 33 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()
|
D | reduction_ops.h | 64 bool keep_dims_;
|