- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 76 for varhandle_op (0.51 sec)
-
tensorflow/c/experimental/ops/gen/cpp/cpp_generator_test.cc
"IdentityN", // Variadic input+output "SparseSoftmaxCrossEntropyWithLogits", // 2 outputs "AccumulatorApplyGradient", // 0 outputs "VarHandleOp", // type, shape, list(string) attrs "RestoreV2", // Variadic output-only, list(type) attr }; cpp::CppConfig cpp_config(category, name_space);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 18 17:02:28 UTC 2022 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/mlir_import_options.h
bool unconditionally_use_set_output_shapes = false; // Apply default attributes from the op definition to the loaded op. bool add_default_attributes = true; // If set, promote tf.VarHandleOp to resource arguments for all functions. bool lift_variables = true; // Keeps the variables in initializers before lifting variables (when // `lift_variables == true`) or newly adding variable initialization patterns
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 20 13:19:26 UTC 2023 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_value_typed_analyzer.cc
auto* operation = resource.getDefiningOp(); if (operation && isa<TF::VarHandleOp>(operation)) { mutable_variables_.insert(GetResourceKey(operation)); } } bool ResourceAnalyzer::IsPotentiallyWritten(Value resource) const { assert(IsResource(resource)); auto* operation = resource.getDefiningOp(); if (operation && isa<TF::VarHandleOp>(operation)) return mutable_variables_.contains(GetResourceKey(operation));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/c/experimental/ops/resource_variable_ops.h
#include "tensorflow/c/eager/abstract_context.h" #include "tensorflow/c/eager/abstract_tensor_handle.h" namespace tensorflow { namespace ops { // Creates a handle to a Variable resource. Status VarHandleOp(AbstractContext* ctx, AbstractTensorHandle** resource, DataType dtype, const PartialTensorShape shape, const char* container = "", const char* shared_name = "",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 10 19:11:36 UTC 2022 - 2.5K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/ops/variable_ops.h
#include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/platform/status.h" namespace tensorflow { namespace internal { // Executes a VarHandleOp using `ctx`, and fills `handle` with the DT_RESOURCE // TensorHandle associated with the variable. This is equivalent to creating an // unitialized TF2 tf.Variable.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 30 21:44:45 UTC 2022 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/freeze_global_tensors.cc
Operation *op = *latticeElement->getValue().ops.begin(); GlobalTensorOp globalTensor = llvm::dyn_cast<GlobalTensorOp>(op); if (!globalTensor) continue; // happens if the name is e.g. in a VarHandleOp. if (globalTensor.getIsMutable()) { freezeable[val] = false; continue; } freezeable[val] = true; // Verify users are supported kind.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/cpp/golden/testing_ops.h.golden
//
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 16 19:04:03 UTC 2023 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/node_expansion_pass.cc
static const char* kOpsToSkip[] = { "IdentityOp", "NoOp", // b/174596063 "OptionalHasValue", // b/173136483 "OptionalGetValue", // b/173136483 "VarHandleOp", // b/176819198 }; for (const char* skip : kOpsToSkip) { if (absl::StartsWith(orig_op->op_name(), skip)) return absl::OkStatus(); } tf_core_op_expansion_node_counter->GetCell()->IncrementBy(1);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Feb 25 16:22:36 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_duplicate_resource_ops.cc
auto& island_block = island_op.getBody().front(); if (++island_block.begin() != --island_block.end()) return nullptr; Operation* resource_op = &island_block.front(); if (llvm::isa<TF::VarHandleOp, TF::HashTableOp, TF::HashTableV2Op, TF::MutableHashTableV2Op>(resource_op)) { return resource_op; } return nullptr; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 26 04:26:16 UTC 2023 - 4.7K bytes - Viewed (0) -
tensorflow/cc/experimental/libtf/tests/variable_test.cc
{ AbstractTensorHandle* var_ptr = nullptr; PartialTensorShape scalar_shape; TF_EXPECT_OK( PartialTensorShape::MakePartialShape<int32_t>({}, 0, &scalar_shape)); TF_EXPECT_OK(tensorflow::ops::VarHandleOp(ctx_.get(), &var_ptr, DT_FLOAT, scalar_shape)); var.reset(var_ptr); } // Assign a value. auto x = CreateScalarTensor<float, TF_FLOAT>(2.0f); TF_EXPECT_OK(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 19 21:44:52 UTC 2023 - 4.5K bytes - Viewed (0)