Home
last modified time | relevance | path

Searched refs:VarHandleOp (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_VarHandleOp.pbtxt2 graph_op_name: "VarHandleOp"
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_VarHandleOp.pbtxt2 graph_op_name: "VarHandleOp"
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model_test.cc130 ops::VarHandleOp(scope.WithOpName("var"), DataType::DT_FLOAT, {}); in TestFreezeGraphWithoutDependentVariables()
175 ops::VarHandleOp(scope.WithOpName("var"), DataType::DT_FLOAT, {}); in TestFreezeGraphWithDependentVariables()
226 ops::VarHandleOp(scope.WithOpName("var"), DataType::DT_FLOAT, {}); in TestFreezeGraphWithAndWithoutDependentVariables()
231 ops::VarHandleOp(scope.WithOpName("var_1"), DataType::DT_FLOAT, {}); in TestFreezeGraphWithAndWithoutDependentVariables()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_VarHandleOp.pbtxt2 graph_op_name: "VarHandleOp"
Dapi_def_ResourceScatterNdAdd.pbtxt6 A resource handle. Must be from a VarHandleOp.
Dapi_def_ResourceScatterNdUpdate.pbtxt6 A resource handle. Must be from a VarHandleOp.
Dapi_def_ResourceScatterNdSub.pbtxt6 A resource handle. Must be from a VarHandleOp.
/external/tensorflow/tensorflow/compiler/jit/
Dresource_operation_safety_analysis_test.cc44 ops::VarHandleOp(scope.WithOpName("Var" + id), DT_FLOAT, TensorShape({})); in MakeRead()
52 ops::VarHandleOp(scope.WithOpName("Var" + id), DT_FLOAT, TensorShape({})); in MakeWrite()
62 ops::VarHandleOp(scope.WithOpName("Var" + id), DT_FLOAT, TensorShape({})); in MakeModify()
Dxla_fusion_optimizer_test.cc187 ops::VarHandleOp(root.WithOpName("Var"), DT_FLOAT, TensorShape({})); in TEST_F()
Dbuild_xla_ops_pass_test.cc109 Output var_handle = ops::VarHandleOp(scope.WithOpName("Var_" + id), DT_FLOAT, in MakeWrite()
Dmark_for_compilation_pass_test.cc522 ops::VarHandleOp(scope.WithOpName("Var" + id), DT_FLOAT, TensorShape({})); in MakeRead()
533 ops::VarHandleOp(scope.WithOpName("Var" + id), DT_FLOAT, TensorShape({})); in MakeWrite()
/external/tensorflow/tensorflow/go/op/
Dscope_test.go77 variable = VarHandleOp(s, tf.Int32, tf.ScalarShape())
Dgradients_test.go224 variable = VarHandleOp(s, tf.Int32, tf.ScalarShape())
Dwrappers.go33350 func VarHandleOp(scope *Scope, dtype tf.DataType, shape tf.Shape, optional ...VarHandleOpAttr) (res… func
/external/tensorflow/tensorflow/core/grappler/costs/graph_properties_testdata/
Dwhile_loop_var_handle_op.pbtxt21 op: "VarHandleOp"
Dloops_and_resource_vars.pbtxt24 op: "VarHandleOp"
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_if_op_test.cc184 auto var = ops::VarHandleOp(root.WithOpName("var"), DT_INT32, {}); in TEST()
Dshape_refiner_test.cc1562 auto x1 = ops::VarHandleOp(root, DataType::DT_FLOAT, TensorShape({128, 256})); in TEST_F()
1563 auto x2 = ops::VarHandleOp(root, DataType::DT_DOUBLE, TensorShape({1024})); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_test.cc793 auto var = ops::VarHandleOp(scope.WithOpName("Variable"), DT_INT32, in TEST()
934 auto var = ops::VarHandleOp(scope.WithOpName("Variable"), DT_INT32, in TEST()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc502 ops::VarHandleOp(s.WithOpName("b"), DataType::DT_FLOAT, {256, 512}); in CreateGrapplerItemWithResourceMemory()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt4256 name: "VarHandleOp"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt4256 name: "VarHandleOp"
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt43463 name: "VarHandleOp"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v1.pbtxt88002 name: "VarHandleOp"
Dops_history.v2.pbtxt87375 name: "VarHandleOp"