Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/kernel_tests/
Ddense_update_ops_no_tsan_test.py79 def run_assign(assign_op): argument
80 self.evaluate(assign_op)
84 target=run_assign, args=(assign_op,)) for assign_op in assigns
146 def run_assign(assign_op): argument
147 self.evaluate(assign_op)
151 target=run_assign, args=(assign_op,)) for assign_op in assigns
/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/
Dvariable_ops.cc74 ImmediateOpPtr assign_op(ctx->CreateOperation()); in AssignVariable() local
75 TF_RETURN_IF_ERROR(assign_op->Reset("AssignVariableOp", nullptr)); in AssignVariable()
76 TF_RETURN_IF_ERROR(assign_op->SetAttrType("dtype", dtype)); in AssignVariable()
77 TF_RETURN_IF_ERROR(assign_op->AddInput(variable_handle)); in AssignVariable()
78 TF_RETURN_IF_ERROR(assign_op->AddInput(value)); in AssignVariable()
81 TF_RETURN_IF_ERROR(assign_op->Execute({}, &num_retvals)); in AssignVariable()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_variables.cc71 TF::AssignVariableOp assign_op, in GetAssignVariableOp() argument
74 assign_op.getLoc(), in GetAssignVariableOp()
75 GetResourceIDAsI32(variable_id, assign_op.getLoc(), builder), in GetAssignVariableOp()
76 assign_op.value()); in GetAssignVariableOp()
108 auto assign_op = llvm::cast<TF::AssignVariableOp>(op); in matchAndRewrite() local
109 tfl_var_op = GetAssignVariableOp(variable_id, assign_op, builder); in matchAndRewrite()
/external/eigen/Eigen/src/SparseCore/
DSparseAssign.h37 …internal::Assignment<Derived,OtherDerived,internal::assign_op<Scalar,typename OtherDerived::Scalar…
38 … ::run(derived(), other.derived(), internal::assign_op<Scalar,typename OtherDerived::Scalar>());
130 …static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<typename DstXprT…
142 …if(internal::is_same<Functor,internal::assign_op<typename DstXprType::Scalar,typename SrcXprType::…
159 struct Assignment<DstXprType, Solve<DecType,RhsType>, internal::assign_op<Scalar,Scalar>, Sparse2Sp…
162 …static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<Scalar,Scalar> &)
185 …alar,Options,StorageIndex> &dst, const SrcXprType &src, const internal::assign_op<typename DstXprT…
201 …arseMatrixBase<DstDerived> &dst, const SrcXprType &src, const internal::assign_op<typename DstXprT…
DSparseProduct.h102 struct Assignment<DstXprType, Product<Lhs,Rhs,AliasFreeProduct>, internal::assign_op<typename DstXp…
105 …static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<typename DstXprT…
DSparseSelfAdjointView.h226 typedef internal::assign_op<typename DstXprType::Scalar,typename SrcXprType::Scalar> AssignOpType;
632 …tXprType, SparseSymmetricPermutationProduct<MatrixType,Mode>, internal::assign_op<Scalar,typename …
637 …x<Scalar,Options,DstIndex> &dst, const SrcXprType &src, const internal::assign_op<Scalar,typename …
646 …intView<DestType,DestMode>& dst, const SrcXprType &src, const internal::assign_op<Scalar,typename …
/external/eigen/Eigen/src/Core/
DSolve.h137 struct Assignment<DstXprType, Solve<DecType,RhsType>, internal::assign_op<Scalar,Scalar>, Dense2Den…
140 …static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<Scalar,Scalar> &)
153 struct Assignment<DstXprType, Solve<Transpose<const DecType>,RhsType>, internal::assign_op<Scalar,S…
156 …static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<Scalar,Scalar> &)
170 internal::assign_op<Scalar,Scalar>, Dense2Dense>
173 …static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<Scalar,Scalar> &)
DAssign_MKL.h84 …tXprType, CwiseUnaryOp<scalar_##EIGENOP##_op<EIGENTYPE>, SrcXprNested>, assign_op<EIGENTYPE,EIGENT…
87 …static void run(DstXprType &dst, const SrcXprType &src, const assign_op<EIGENTYPE,EIGENTYPE> &/*fu…
143 …const CwiseNullaryOp<internal::scalar_constant_op<EIGENTYPE>,Plain> >, assign_op<EIGENTYPE,EIGENTY…
147 …static void run(DstXprType &dst, const SrcXprType &src, const assign_op<EIGENTYPE,EIGENTYPE> &/*fu…
DAssignEvaluator.h715 void resize_if_allowed(DstXprType &dst, const SrcXprType& src, const internal::assign_op<T1,T2> &/*…
747 …call_dense_assignment_loop(dst, src, internal::assign_op<typename DstXprType::Scalar,typename SrcX…
782 call_assignment(dst, src, internal::assign_op<typename Dst::Scalar,typename Src::Scalar>());
788 call_assignment(dst, src, internal::assign_op<typename Dst::Scalar,typename Src::Scalar>());
842 …call_assignment_no_alias(dst, src, internal::assign_op<typename Dst::Scalar,typename Src::Scalar>(…
860 …call_assignment_no_alias_no_transpose(dst, src, internal::assign_op<typename Dst::Scalar,typename …
891 …INLINE void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<typename DstXprT…
DNoAlias.h42 …call_assignment_no_alias(m_expression, other.derived(), internal::assign_op<Scalar,typename OtherD…
/external/tensorflow/tensorflow/python/framework/
Dauto_control_deps_test.py77 assign_op = gen_resource_variable_ops.assign_variable_op(
81 self.assertIn(read_op1, assign_op.control_inputs)
82 self.assertIn(read_op2, assign_op.control_inputs)
92 assign_op = gen_resource_variable_ops.assign_variable_op(
99 self.assertIn(assign_op, read_op1.control_inputs)
100 self.assertIn(assign_op, read_op2.control_inputs)
206 assign_op = get_op("AssignVariableOp", "")
212 self.assertIn(read1, assign_op.control_inputs)
213 self.assertIn(read2, assign_op.control_inputs)
214 self.assertIn(functional_op, assign_op.control_inputs)
[all …]
/external/eigen/Eigen/src/Core/functors/
DAssignmentFunctors.h21 template<typename DstScalar,typename SrcScalar> struct assign_op { struct
23 EIGEN_EMPTY_STRUCT_CTOR(assign_op) in EIGEN_EMPTY_STRUCT_CTOR() argument
32 template<typename DstScalar> struct assign_op<DstScalar,void> {}; argument
35 struct functor_traits<assign_op<DstScalar,SrcScalar> > {
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_merge_variables_with_execute.cc250 auto assign_op = llvm::dyn_cast<TF::AssignVariableOp>(*result.user_begin()); in BuildVariableAccessInfo() local
251 if (!assign_op) continue; in BuildVariableAccessInfo()
252 auto resource = assign_op.resource(); in BuildVariableAccessInfo()
264 info.assign = assign_op; in BuildVariableAccessInfo()
265 if (!last_assign || last_assign->isBeforeInBlock(assign_op)) { in BuildVariableAccessInfo()
266 last_assign = assign_op; in BuildVariableAccessInfo()
268 all_assigns.insert(assign_op); in BuildVariableAccessInfo()
/external/eigen/Eigen/src/IterativeLinearSolvers/
DSolveWithGuess.h96 struct Assignment<DstXprType, SolveWithGuess<DecType,RhsType,GuessType>, internal::assign_op<Scalar…
99 …static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<Scalar,Scalar> &)
/external/tensorflow/tensorflow/python/saved_model/
Dloader_test.py88 assign_op = control_flow_ops.group(state_ops.assign(y, 7))
94 sess, ["foo_graph"], sig_def_map, init_op=assign_op)
97 sess, ["foo_graph"], sig_def_map, main_op=assign_op)
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dautocast_variable.py226 assign_op = update_fn(value, use_locking, name, False)
232 var._op = assign_op # pylint:disable=protected-access
234 return assign_op
/external/eigen/Eigen/src/Geometry/
DHomogeneous.h351 struct Assignment<DstXprType, Homogeneous<ArgType,Vertical>, internal::assign_op<Scalar,typename Ar…
354 …static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<Scalar,typename …
368 struct Assignment<DstXprType, Homogeneous<ArgType,Horizontal>, internal::assign_op<Scalar,typename …
371 …static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<Scalar,typename …
/external/eigen/Eigen/src/SparseQR/
DSparseQR.h708 struct Assignment<DstXprType, SparseQRMatrixQReturnType<SparseQRType>, internal::assign_op<typename…
713 …static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<Scalar,Scalar> &…
724 struct Assignment<DstXprType, SparseQRMatrixQReturnType<SparseQRType>, internal::assign_op<typename…
729 …static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<Scalar,Scalar> &…
/external/tensorflow/tensorflow/python/training/tracking/
Dutil.py1656 assign_op = self.save_counter.assign_add(1, read_value=True)
1658 self._save_assign_op = data_structures.NoDependency(assign_op)
1662 checkpoint_number = assign_op.numpy()
2093 assign_op = self.save_counter.assign_add(1, read_value=True)
2095 self._save_assign_op = data_structures.NoDependency(assign_op)
2099 checkpoint_number = assign_op.numpy()
/external/tensorflow/tensorflow/python/distribute/
Dmoving_averages_test.py96 var, assign_op = distribution.extended.call_for_each_replica(replica_fn)
99 self.evaluate(distribution.experimental_local_results(assign_op))
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass_test.cc112 ops::AssignVariableOp assign_op(scope.WithOpName("Assignee_" + id), in MakeWrite() local
114 return assign_op.operation.node(); in MakeWrite()
/external/eigen/Eigen/src/QR/
DCompleteOrthogonalDecomposition.h531 …rType, Inverse<CompleteOrthogonalDecomposition<MatrixType> >, internal::assign_op<typename DstXprT…
535 …static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<typename DstXprT…
/external/tensorflow/tensorflow/core/common_runtime/
Dquantize_training.cc301 Node* assign_op; in AddRestoreVariableSubgraphs() local
305 .Finalize(graph, &assign_op)); in AddRestoreVariableSubgraphs()
308 graph->AddControlEdge(assign_op, restore_all); in AddRestoreVariableSubgraphs()
/external/eigen/Eigen/src/LU/
DInverseImpl.h290 struct Assignment<DstXprType, Inverse<XprType>, internal::assign_op<typename DstXprType::Scalar,typ…
293 …static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<typename DstXprT…
DPartialPivLU.h568 struct Assignment<DstXprType, Inverse<PartialPivLU<MatrixType> >, internal::assign_op<typename DstX…
572 …static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<typename DstXprT…

12