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.py72 def run_assign(assign_op): argument
73 self.evaluate(assign_op)
77 target=run_assign, args=(assign_op,)) for assign_op in assigns
133 def run_assign(assign_op): argument
134 self.evaluate(assign_op)
138 target=run_assign, args=(assign_op,)) for assign_op in assigns
/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…
DTriangularMatrix.h560 …internal::call_assignment_no_alias(derived(), other.derived(), internal::assign_op<Scalar,typename…
814 …call_triangular_assignment_loop<Mode,SetOpposite>(dst, src, internal::assign_op<typename DstXprTyp…
943 struct Assignment<DstXprType, Product<Lhs,Rhs,DefaultProduct>, internal::assign_op<Scalar,typename …
946 …static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<Scalar,typename …
/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/contrib/framework/python/ops/
Dvariables.py563 assign_op = control_flow_ops.group(*assign_ops)
564 return assign_op, feed_dict
583 assign_op, feed_dict = assign_from_values(var_names_to_values)
585 return session.run(assign_op, feed_dict)
698 assign_op = control_flow_ops.group(*assign_ops)
699 return assign_op, feed_dict
/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/contrib/distribute/python/
Dmoving_averages_test.py82 var, assign_op = distribution.extended.call_for_each_replica(replica_fn)
85 sess.run(distribution.experimental_local_results(assign_op))
/external/tensorflow/tensorflow/python/saved_model/
Dloader_test.py81 assign_op = control_flow_ops.group(state_ops.assign(y, 7))
87 sess, ["foo_graph"], sig_def_map, init_op=assign_op)
90 sess, ["foo_graph"], sig_def_map, main_op=assign_op)
/external/tensorflow/tensorflow/python/training/tracking/
Dutil.py1359 assign_op = self.save_counter.assign_add(1, read_value=True)
1361 self._save_assign_op = data_structures.NoDependency(assign_op)
1365 checkpoint_number = assign_op.numpy()
1663 assign_op = self.save_counter.assign_add(1, read_value=True)
1665 self._save_assign_op = data_structures.NoDependency(assign_op)
1669 checkpoint_number = assign_op.numpy()
/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/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/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass_test.cc111 ops::AssignVariableOp assign_op(scope.WithOpName("Assignee_" + id), in MakeWrite() local
113 return assign_op.operation.node(); in MakeWrite()
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
Dkeras_tpu_variables.py190 assign_op = gen_resource_variable_ops.assign_variable_op(
194 return assign_op
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Doptimize_dataset_test.py227 assign_op = variable.assign_add(1)
253 with ops.control_dependencies([assign_op]):
/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…
DColPivHouseholderQR.h617 struct Assignment<DstXprType, Inverse<ColPivHouseholderQR<MatrixType> >, internal::assign_op<typena…
621 …static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<typename DstXprT…
/external/tensorflow/tensorflow/core/graph/
Dquantize_training.cc303 Node* assign_op; in AddRestoreVariableSubgraphs() local
307 .Finalize(graph, &assign_op)); in AddRestoreVariableSubgraphs()
310 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