Searched defs:RestoreOp (Results 1 – 5 of 5) sorted by relevance
257 struct RestoreOp { struct258 RestoreOp(OpKernelContext* context, int idx, const string& tensor_name, in RestoreOp() function274 bool should_run_in_pool(BundleReader* reader) const { in should_run_in_pool()286 void run_with_new_reader() { in run_with_new_reader()296 Status run(BundleReader* reader) { in run()352 OpKernelContext* context;353 int idx;354 string tensor_name;355 string shape_and_slice;356 string reader_prefix;[all …]
27 class RestoreOp : public OpKernel { class29 explicit RestoreOp(OpKernelConstruction* context) : OpKernel(context) { in RestoreOp() function in tensorflow::RestoreOp
1787 unsigned RestoreOp, MoveReg; in restoreCRs() local
2280 unsigned RestoreOp, MoveReg; in restoreCRs() local
2520 unsigned RestoreOp = PPC::MTOCRF; in restoreCRs() local