- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for Platen (0.07 sec)
-
tensorflow/cc/experimental/libtf/function.cc
// TODO(b/190203981): Move to a separate nest-like library. void Flatten(const TaggedValue& value, std::vector<AbstractTensorHandle*>* flat_args) { if (value.type() == TaggedValue::Type::TENSOR) { flat_args->emplace_back(value.tensor().get()); } else if (value.type() == TaggedValue::Type::TUPLE) { for (const auto& t : value.tuple()) { Flatten(t, flat_args); } } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 04 19:49:06 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.cc
// Returning null tensors from a tf.function is not supported, so we keep // track of indices in the model's outputs are nullptr in this set. // The FunctionDef only outputs the non-null tensors. We later pad the // function op outputs to have nullptrs at the `null_indices`. absl::flat_hash_set<int> null_indices; { AbstractContextPtr func_ctx(BuildFunction(fn_name));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 13:57:45 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/jit/cluster_scoping_pass.cc
// Node_Y will receive both scopes "unstage" and "stage", while Node_X receives // only scope "stage". The semantic of scope "unstage" is preserved although // scope "stage" is later appended. As a result, Node_X and Node_Y will be put // into different clusters. // // Unstage -> Node_Y (scope "unstage & stage") // | // V
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/reduce.h
// Generate a Max and an ArgMax of as the mhlo op returns both while in TF // we have separate ops for them. If only one of them is used then the other // one will be garbage collected later. if (!mlir::isa<ShapedType>(operand.getType())) return failure(); auto operand_type = mlir::cast<ShapedType>(operand.getType()); if (operand_type.getElementType().isInteger(1)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/constant_fold_utils.cc
return false; } // If any of the result types are variants, don't try to constant fold them. // This creates opaque variant constants which lose information and would // require "raising" later. for (const Type type : inst->getResultTypes()) { if (const TensorType tensor_type = mlir::dyn_cast<TensorType>(type)) { if (mlir::isa<VariantType>(tensor_type.getElementType())) { return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.3K bytes - Viewed (0)