Home
last modified time | relevance | path

Searched defs:assign_var (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_parallel_execute_sink_resource_write.cc53 auto assign_var = dyn_cast<TF::AssignVariableOp>(use.getOwner()); in GetSingleUseResourceWrite() local
91 TF::AssignVariableOp assign_var = in SinkResourceWritesIntoParallelExecute() local
Dtpu_resource_partitioning.cc89 auto assign_var = in PartitionResourceReadsWrites() local
Dtpu_resource_read_for_write.cc53 auto assign_var = dyn_cast<TF::AssignVariableOp>(result_user); in GetResourceWriteResult() local
Dtpu_sharding_identification_pass.cc197 if (auto assign_var = llvm::dyn_cast<TF::AssignVariableOp>(user)) in GetXlaShardingFromResult() local
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass_test.cc201 auto assign_var = NodeWith(Op("AssignVariableOp"), Inputs(_, Out(merge))); in TEST_F() local
233 auto assign_var = in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dinitialize_variables.cc92 else if (auto assign_var = llvm::dyn_cast_or_null<TF::AssignVariableOp>(op)) in GetGlobalTensorOp() local