Searched defs:keep_dims_ (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dreduction_ops.h65 bool keep_dims_; member in class:tensorflow::XlaReductionOp
/external/tensorflow/tensorflow/core/kernels/
H A Dreduce_join_op.cc119 OP_REQUIRES_OK(ctx, ctx->GetAttr("keep_dims", &keep_dims_));
156 GetOutputShape(index_is_reduced, input_shape, keep_dims_);
181 bool keep_dims_; local
H A 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_; member in class:tensorflow::ReductionOp
H A Dsparse_reduce_op.cc157 OP_REQUIRES_OK(ctx, ctx->GetAttr("keep_dims", &keep_dims_));
178 sp, reduction_axes_t->flat<int32>(), keep_dims_);
230 bool keep_dims_; member in class:tensorflow::SparseReduceOp
251 OP_REQUIRES_OK(ctx, ctx->GetAttr("keep_dims", &keep_dims_));
266 sp, reduction_axes_t->flat<int32>(), keep_dims_);
300 if (keep_dims_) {
324 bool keep_dims_; member in class:tensorflow::SparseReduceSparseOp

Completed in 1655 milliseconds