Searched refs:keep_dims_ (Results 1 – 9 of 9) sorted by relevance
| /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/lite/delegates/xnnpack/ |
| D | fully_connected_tester.h | 68 keep_dims_ = keep_dims; in KeepDims()
72 inline bool KeepDims() const { return keep_dims_; } in KeepDims()
111 bool keep_dims_ = false; variable
|
| D | reduce_tester.h | 60 keep_dims_ = keep_dims; in KeepDims()
64 inline bool KeepDims() const { return keep_dims_; } in KeepDims()
110 bool keep_dims_ = true; variable
|
| /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()
180 bool keep_dims_; member in tensorflow::ReduceJoinOp
|
| D | reduction_ops_common.h | 138 OP_REQUIRES_OK(ctx, ctx->GetAttr("keep_dims", &keep_dims_));
148 OP_REQUIRES_OK(ctx, helper.Simplify(data, axes, keep_dims_));
250 bool 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()
98 } else if (keep_dims_) { in Compile()
126 auto result = keep_dims_ ? xla::Reshape(finalized, final_shape) : finalized; in Compile()
|
| D | reduction_ops.h | 64 bool keep_dims_;
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 5466 op_attrs.keep_dims_ = keep_dims; in CreateReduceOp()
|