Home
last modified time | relevance | path

Searched refs:assign_variable_op (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dauto_control_deps_test.py94 gen_resource_variable_ops.assign_variable_op(v.handle, 1)
96 gen_resource_variable_ops.assign_variable_op(v.handle, 2))
121 gen_resource_variable_ops.assign_variable_op(v.handle, i))
138 gen_resource_variable_ops.assign_variable_op(v.handle, v + 1)
151 assign_op = gen_resource_variable_ops.assign_variable_op(
166 assign_op = gen_resource_variable_ops.assign_variable_op(
184 gen_resource_variable_ops.assign_variable_op(v.handle, v + 1)
186 assign_op2 = gen_resource_variable_ops.assign_variable_op(
215 assign_op1 = gen_resource_variable_ops.assign_variable_op(
217 assign_op2 = gen_resource_variable_ops.assign_variable_op(
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dvariable_ops_test.py243 resource_variable_ops.assign_variable_op(
258 resource_variable_ops.assign_variable_op(
273 resource_variable_ops.assign_variable_op(
288 resource_variable_ops.assign_variable_op(
303 resource_variable_ops.assign_variable_op(
318 resource_variable_ops.assign_variable_op(
331 resource_variable_ops.assign_variable_op(
344 resource_variable_ops.assign_variable_op(
357 resource_variable_ops.assign_variable_op(
372 resource_variable_ops.assign_variable_op(
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/variables/
H A Dresource_variable_ops_test.py88 resource_variable_ops.assign_variable_op(
91 resource_variable_ops.assign_variable_op(handle,
95 resource_variable_ops.assign_variable_op(handle,
130 resource_variable_ops.assign_variable_op(handle, 1)
235 resource_variable_ops.assign_variable_op(
244 resource_variable_ops.assign_variable_op(
278 self.evaluate(resource_variable_ops.assign_variable_op(
288 self.evaluate(resource_variable_ops.assign_variable_op(
297 create = resource_variable_ops.assign_variable_op(
303 write = resource_variable_ops.assign_variable_op(
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dresource_op_lifting.cc148 if (auto assign_variable_op = dyn_cast<TF::AssignVariableOp>(&op)) { in ForwardStoreToLoad() local
149 Value resource = assign_variable_op.resource(); in ForwardStoreToLoad()
154 resource_handle_to_last_store_op[resource] = assign_variable_op; in ForwardStoreToLoad()
435 for (auto assign_variable_op : llvm::make_early_inc_range(assign_ops)) { in AppendResourceStoreValueToReturn() local
436 Value resource = assign_variable_op.resource(); in AppendResourceStoreValueToReturn()
444 new_return_operands[result_index] = assign_variable_op.value(); in AppendResourceStoreValueToReturn()
445 assign_variable_op.erase(); in AppendResourceStoreValueToReturn()
768 for (auto assign_variable_op : llvm::make_early_inc_range(assign_ops)) { in LiftArgRetResourcesForFunction() local
769 Value resource = assign_variable_op.resource(); in LiftArgRetResourcesForFunction()
773 handle_updated_arg_value(arg.getArgNumber(), assign_variable_op.value()); in LiftArgRetResourcesForFunction()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dtpu_values.py302 gen_resource_variable_ops.assign_variable_op)(
374 gen_resource_variable_ops.assign_variable_op)(self, *args, **kwargs)
405 gen_resource_variable_ops.assign_variable_op)
461 gen_resource_variable_ops.assign_variable_op)(
562 gen_resource_variable_ops.assign_variable_op)(var, *args, **kwargs)
H A Dtpu_replicated_variable.py233 gen_resource_variable_ops.assign_variable_op)(
/aosp_15_r20/external/tensorflow/tensorflow/python/training/saving/
H A Dsaveable_object_util_test.py242 gen_resource_variable_ops.assign_variable_op(self.resource_handle,
254 gen_resource_variable_ops.assign_variable_op(self.resource_handle, value)
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dspmd_expansion.cc178 if (auto assign_variable_op = in IsValueUsedByAssignVariableOp() local
182 GetForwardedDTensorLayoutInput(assign_variable_op.resource()) in IsValueUsedByAssignVariableOp()
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Ddef_function.py343 self._initializer_op = resource_variable_ops.assign_variable_op(
350 resource_variable_ops.assign_variable_op(
360 resource_variable_ops.assign_variable_op(
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/experimental/
H A Dgraph_building_test.py52 resource_variable_ops.assign_variable_op(
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dresource_variable_ops.py311 return gen_resource_variable_ops.assign_variable_op(
943 assign_op = gen_resource_variable_ops.assign_variable_op(
1862 gen_resource_variable_ops.assign_variable_op(
1890 gen_resource_variable_ops.assign_variable_op(handle, initial_value)
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/parallel_device/
H A Dparallel_device_test.py319 gen_resource_variable_ops.assign_variable_op(h1, constant_op.constant(3.))
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dload_test.py1875 resource_variable_ops.assign_variable_op(
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
H A Dcontrol_flow_ops_py_test.py509 assign = resource_variable_ops.assign_variable_op(rv.handle, False)