- Sort Score
- Result 10 results
- Languages All
Results 11 - 14 of 14 for var_handle_op (0.6 sec)
-
tensorflow/c/eager/c_api_test_util.cc
TFE_TensorHandle* TestVariable(TFE_Context* ctx, float value, const tensorflow::string& device_name) { TF_Status* status = TF_NewStatus(); // Create the variable handle. TFE_Op* op = TFE_NewOp(ctx, "VarHandleOp", status); if (TF_GetCode(status) != TF_OK) return nullptr; TFE_OpSetAttrType(op, "dtype", TF_FLOAT); TFE_OpSetAttrShape(op, "shape", {}, 0, status); TFE_OpSetAttrString(op, "container", "localhost", 0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mark_ops_for_outside_compilation.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 16:22:32 UTC 2024 - 29.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/control_flow.mlir
func.return %arg : tensor<!tf_type.resource<tensor<i32>>> } // CHECK-LABEL: func @side_effect_while_test func.func @side_effect_while_test() -> (tensor<i32>) { %0 = "tf.VarHandleOp"() {device = "/device:CPU:0", container = "c", shared_name = "v"} : () -> tensor<!tf_type.resource<tensor<i32>>> // CHECK: [[while_res:%.]]:2 = tfrt.while {{%.*}} @"side_effect_while_body_add2/tfrt_body_1"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 00:40:32 UTC 2024 - 17.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/passes.h
// aliasing output arguments. std::unique_ptr<OperationPass<ModuleOp>> CreatePromoteResourcesToArgsPass( llvm::ArrayRef<std::string> functions = {}); // Creates a pass that promotes tf.VarHandleOp to resource arguments for all // functions. std::unique_ptr<OperationPass<ModuleOp>> CreatePromoteVarHandlesToArgsPass(); // Creates a pass that converts readonly reference variables to the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 31.8K bytes - Viewed (0)