Searched refs:binary_output_ (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | count_ops.cc | 105 OP_REQUIRES_OK(context, context->GetAttr("binary_output", &binary_output_)); in DenseCount()
160 if (binary_output_) { in Compute()
184 bool binary_output_; member in tensorflow::DenseCount
193 OP_REQUIRES_OK(context, context->GetAttr("binary_output", &binary_output_)); in SparseCount()
289 if (binary_output_) { in Compute()
311 bool binary_output_; member in tensorflow::SparseCount
321 OP_REQUIRES_OK(context, context->GetAttr("binary_output", &binary_output_)); in RaggedCount()
376 if (binary_output_) { in Compute()
398 bool binary_output_; member in tensorflow::RaggedCount
|
| H A D | bincount_op.cc | 261 OP_REQUIRES_OK(ctx, ctx->GetAttr("binary_output", &binary_output_)); in DenseBincountOp()
302 if (binary_output_) { in Compute()
321 if (binary_output_) { in Compute()
335 bool binary_output_; member in tensorflow::DenseBincountOp
376 OP_REQUIRES_OK(ctx, ctx->GetAttr("binary_output", &binary_output_)); in SparseBincountOp()
407 if (binary_output_) { in Compute()
438 if (binary_output_) { in Compute()
453 bool binary_output_; member in tensorflow::SparseBincountOp
474 OP_REQUIRES_OK(ctx, ctx->GetAttr("binary_output", &binary_output_)); in RaggedBincountOp()
523 if (binary_output_) { in Compute()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | bincount_op.cc | 36 (void)ctx->GetAttr("binary_output", &binary_output_); in DenseBincountOp()
40 bool binary_output_ = false; member in tensorflow::__anon974eb8cc0111::DenseBincountOp
124 if (has_weights && !binary_output_) { in Compile()
134 if (has_weights && !binary_output_) { in Compile()
145 if (!binary_output_) { in Compile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/security/advisory/ |
| H A D | tfsa-2021-002.md | 29 if (binary_output_) {
|