- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for validate_shape (0.24 sec)
-
tensorflow/c/experimental/ops/resource_variable_ops.cc
TF_RETURN_IF_ERROR(op_ptr->AddInput(resource)); TF_RETURN_IF_ERROR(op_ptr->AddInput(value)); TF_RETURN_IF_ERROR(op_ptr->SetAttrBool("validate_shape", validate_shape)); int num_retvals = 0; std::vector<AbstractTensorHandle*> dummy_outputs; return op_ptr->Execute(absl::MakeSpan(dummy_outputs), &num_retvals); } // Op: DestroyResourceOp() // Summary: Deletes the resource specified by the handle. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 10 19:11:36 UTC 2022 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/testdata/func_with_dead_ops.mlir
"tf.AssignVariableOp"(%arg3, %13#10) <{validate_shape = false}> : (tensor<*x!tf_type.resource<tensor<128x1024xf32>>>, tensor<128x1024xf32>) -> () "tf.AssignVariableOp"(%arg5, %13#8) <{validate_shape = false}> : (tensor<*x!tf_type.resource<tensor<1024x1xf32>>>, tensor<1024x1xf32>) -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 13 23:22:50 UTC 2024 - 15.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/insert_restore_op.mlir
// CHECK: "tf.AssignVariableOp"(%[[VAR_HANDLE_0]], %[[RESTORE]]#0) <{validate_shape = false}> : (tensor<!tf_type.resource<tensor<2xf32>>>, tensor<2xf32>) -> () // CHECK: "tf.AssignVariableOp"(%[[VAR_HANDLE_1]], %[[RESTORE]]#1) <{validate_shape = false}> : (tensor<!tf_type.resource<tensor<4xi32>>>, tensor<4xi32>) -> () // Test that the locs are properly set to their shared_names.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/convert_ref_variables.mlir
%1 = "tf.Const"() {value = dense<0> : tensor<i32>} : () -> tensor<i32> %2 = "tf.Assign"(%0, %1) {T = i32, device = "", use_locking = true, validate_shape = true} : (tensor<!tf_type.int32ref>, tensor<i32>) -> tensor<!tf_type.int32ref> func.return } // CHECK-LABEL: @inference func.func @inference() -> tensor<i32> { // CHECK-NOT: tf.VariableV2
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_variables.td
(TF_ReadVariableOp:$result $resource), (TFL_ReadVariableOp $resource), [(HasSupportedElementType $result)]>; def LegalizeAssignVariable : Pat< (TF_AssignVariableOp $resource, $value, $validate_shape), (TFL_AssignVariableOp $resource, $value),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 16 23:20:46 UTC 2022 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/merge_duplicate_resource_ops.mlir
%control_13 = tf_executor.island(%control_12) wraps "tf.AssignVariableOp"(%outputs_9, %outputs_11) {validate_shape = false} : (tensor<!tf_type.resource<tensor<20x4096xf32>>>, tensor<20x4096xf32>) -> () %control_14 = tf_executor.island(%control_13) wraps "tf.NoOp"() : () -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 26 04:26:16 UTC 2023 - 10.5K bytes - Viewed (0) -
tensorflow/c/kernels_experimental.h
// conditions are met: // 1. `validate_shape` is set to `true` // 2. The variable is initialized // 3. The shape of the value tensor doesn't match the shape of the variable // tensor. TF_CAPI_EXPORT extern void TF_AssignVariable( TF_OpKernelContext* ctx, int input_index, int value_index, bool validate_shape, void (*copyFunc)(TF_OpKernelContext* ctx, TF_Tensor* source,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 07 14:44:39 UTC 2023 - 9.4K bytes - Viewed (0) -
tensorflow/c/experimental/ops/resource_variable_ops.h
Status AssignVariableOp(AbstractContext* ctx, AbstractTensorHandle* const resource, AbstractTensorHandle* const value, bool validate_shape = false, const char* name = nullptr, const char* raw_device_name = nullptr); // Deletes the resource specified by the handle. Status DestroyResourceOp(AbstractContext* ctx,
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/compiler/mlir/tensorflow/tests/decompose_resource_ops.mlir
// CHECK: "tf.AssignVariableOp"([[VAR_HANDLE]], [[NEW_VAR]]) <{validate_shape = false}> : (tensor<*x!tf_type.resource<tensor<*xf32>>>, tensor<*xf32>) -> () // CHECK: "tf.AssignVariableOp"([[M_HANDLE]], [[NEW_M]]) <{validate_shape = false}> : (tensor<*x!tf_type.resource<tensor<*xf32>>>, tensor<*xf32>) -> () // CHECK: "tf.AssignVariableOp"([[V_HANDLE]], [[NEW_V]]) <{validate_shape = false}> : (tensor<*x!tf_type.resource<tensor<*xf32>>>, tensor<*xf32>) -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 51.3K bytes - Viewed (0) -
tensorflow/c/kernels_experimental.cc
return nullptr; } } return cc_ctx->input_ref_mutex(input); } void TF_AssignVariable(TF_OpKernelContext* ctx, int input_index, int value_index, bool validate_shape, void (*copyFunc)(TF_OpKernelContext* ctx, TF_Tensor* source, TF_Tensor* dest), TF_Status* status) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:12:29 UTC 2024 - 30.9K bytes - Viewed (0)