- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for GetBoolAttr (0.27 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
if (!parent_op) return false; return HasOutsideCompilationAncestor(parent_op); } Operation* ApplyXlaHostTransferAttr(Operation* op, OpBuilder& builder) { op->setAttr("_xla_has_host_transfer", builder.getBoolAttr(true)); return op; } // Creates a tf._XlaSendFromHost or tf._XlaSendFromHostV2 op. If device ordinal // is present, a tf._XlaSendFromHostV2 op is created instead.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
/*then_branch=*/ mlir::SymbolRefAttr::get(then_branch_op), /*else_branch=*/ mlir::SymbolRefAttr::get(else_branch_op), /*is_stateless=*/rewriter.getBoolAttr(true)); return success(); } private: // When the input tensorlist's size is smaller than the requested size, // then branch is executed.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
// gives mathematically equivalent result. auto max_logits = rewriter.create<TF::MaxOp>(loc, logits, reduce_dim, /*keep_dims=*/rewriter.getBoolAttr(true)); auto shifted_logits = rewriter.create<TF::SubOp>(loc, logits, max_logits); // Exponentiate the inputs. Value exp = rewriter.create<TF::ExpOp>(loc, shifted_logits);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
return vec; } void SetBasicBlockAttributes(OpBuilder& builder, Operation* op) { op->setAttr(kDevice, builder.getStringAttr("")); op->setAttr(kLower, builder.getBoolAttr(true)); } std::vector<Value> ResultsAsVector(Operation* op, int begin, int num) { int end = begin + num; std::vector<Value> vec; vec.reserve(end - begin);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
builder.setInsertionPointAfter(last_store); auto cast = builder.create<TF::CastOp>( last_store.getLoc(), read_type, last_store.getValue(), /*Truncate=*/builder.getBoolAttr(false)); read_variable_op.getValue().replaceAllUsesWith(cast); } else { read_variable_op.getValue().replaceAllUsesWith(last_store.getValue()); } read_variable_op.erase();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
builder.getNamedAttr("then_branch", then_attr), builder.getNamedAttr("else_branch", else_attr), // TODO(b/139667752): Analyze statelessness correctly builder.getNamedAttr("is_stateless", builder.getBoolAttr(false))}; } if (auto* opts = options.AsWhileOptions()) { uint32_t cond_idx = opts->cond_subgraph_index; if (cond_idx >= func_names.size()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0)