Home
last modified time | relevance | path

Searched defs:enqueue_op (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
H A Dpriority_queue_test.py85 def enqueue(enqueue_op): argument
139 def enqueue(enqueue_op): argument
194 def enqueue(enqueue_op): argument
249 def enqueue(enqueue_op): argument
H A Dpadding_fifo_queue_test.py128 def enqueue(enqueue_op): argument
1447 def _blockingEnqueue(self, sess, enqueue_op): argument
H A Dfifo_queue_test.py781 def enqueue(enqueue_op): argument
939 def _blockingEnqueue(self, sess, enqueue_op): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dupdate_tpu_embedding_ops_passes.cc114 std::map<std::string, N>* enqueue_op, in UpdateMapsForModeOverride()
150 const std::map<std::string, N>& enqueue_op, in ComputeEnqueueTrainingStatus()
179 std::map<std::string, Node*> enqueue_op; in GetEnqueueOpsFromGraph() local
222 std::map<std::string, int> enqueue_op; in GetEnqueueOpsFromFunctionDef() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/
H A Dgrappler_item.cc79 for (const string& enqueue_op : queue_runner.enqueue_op_name()) { in EnqueueOpsFanin() local
133 for (const string& enqueue_op : queue_runner.enqueue_op_name()) { in NodesToPreserve() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/training/
H A Dqueue_runner.cc98 for (const string& enqueue_op : enqueue_op_names_) { in Start() local
162 void QueueRunner::Run(Session* sess, const string& enqueue_op) { in Run()
H A Dqueue_runner_test.cc89 for (const std::string& enqueue_op : enqueue_ops) { in BuildQueueRunnerDef() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/random/
H A Drandom_shuffle_queue_test.py104 def enqueue(enqueue_op): argument
1230 def _blockingEnqueue(self, sess, enqueue_op): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dfifo_queue_test.py74 def enqueue(enqueue_op): argument
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/
H A DCLScheduler.cpp211 void CLScheduler::enqueue_op(ICLKernel &kernel, ITensorPack &tensors, bool flush) in enqueue_op() function in arm_compute::CLScheduler
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dqueue_runner_impl.py236 def _run(self, sess, enqueue_op, coord=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_experimental.cc296 TF_Operation* enqueue_op = TF_GraphOperationByName( in TF_EnqueueNamedTensor() local
/aosp_15_r20/external/federated-compute/fcp/protos/
H A Dplan.proto760 string enqueue_op = 1; field
/aosp_15_r20/out/soong/.intermediates/external/federated-compute/federated-compute-cc-proto-lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/fcp/protos/
Dplan.pb.h15015 inline const std::string& AsyncInputFeed::enqueue_op() const { in enqueue_op() function
15046 inline void AsyncInputFeed::set_allocated_enqueue_op(std::string* enqueue_op) { in set_allocated_enqueue_op()