Home
last modified time | relevance | path

Searched refs:enqueue_op (Results 1 – 25 of 41) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
H A Dpadding_fifo_queue_test.py97 enqueue_op = q.enqueue((10.0,))
98 enqueue_op.run()
128 def enqueue(enqueue_op): argument
129 self.evaluate(enqueue_op)
157 for enqueue_op in enqueue_ops:
158 enqueue_op.run()
180 for enqueue_op in enqueue_ops:
181 enqueue_op.run()
201 for enqueue_op in enqueue_ops:
202 self.evaluate(enqueue_op)
[all …]
H A Dfifo_queue_test.py463 enqueue_op = q.enqueue_many((elems,))
467 enqueue_op.run()
480 enqueue_op = q.enqueue((10.0,))
483 enqueue_op.run()
619 enqueue_op = q.enqueue(10.0)
621 enqueue_op = q.enqueue((10.0,))
624 enqueue_op = q.enqueue({})
626 enqueue_op = q.enqueue({"x": 12})
628 enqueue_op = q.enqueue({"f": 10.0, "s": "aa"})
629 enqueue_op = q.enqueue({"f": 10.0})
[all …]
H A Dpriority_queue_test.py85 def enqueue(enqueue_op): argument
86 self.evaluate(enqueue_op)
139 def enqueue(enqueue_op): argument
140 self.evaluate(enqueue_op)
194 def enqueue(enqueue_op): argument
195 self.evaluate(enqueue_op)
249 def enqueue(enqueue_op): argument
250 self.evaluate(enqueue_op)
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/random/
H A Drandom_shuffle_queue_test.py52 enqueue_op = q.enqueue((10.0,))
54 enqueue_op.run()
104 def enqueue(enqueue_op): argument
105 self.evaluate(enqueue_op)
130 for enqueue_op in enqueue_ops:
131 enqueue_op.run()
153 for enqueue_op in enqueue_ops:
154 enqueue_op.run()
170 for enqueue_op in enqueue_ops:
171 self.evaluate(enqueue_op)
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dfifo_queue_test.py30 enqueue_op = q.enqueue((10.0,))
31 enqueue_op.run()
74 def enqueue(enqueue_op): argument
75 sess.run(enqueue_op)
99 for enqueue_op in enqueue_ops:
100 enqueue_op.run()
122 for enqueue_op in enqueue_ops:
123 enqueue_op.run()
140 for enqueue_op in enqueue_ops:
141 sess.run(enqueue_op)
[all …]
/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() argument
137 TF_RET_CHECK(enqueue_op->find(layer_call_index) == enqueue_op->end()) in UpdateMapsForModeOverride()
142 (*enqueue_op)[layer_call_index] = node_identifier; in UpdateMapsForModeOverride()
150 const std::map<std::string, N>& enqueue_op, in ComputeEnqueueTrainingStatus() argument
153 TF_RET_CHECK(enqueue_op.size() == found_recv_op.size()) in ComputeEnqueueTrainingStatus()
155 << "Found " << enqueue_op.size() << " enqueue(s) and " in ComputeEnqueueTrainingStatus()
157 for (const auto& node : enqueue_op) { in ComputeEnqueueTrainingStatus()
179 std::map<std::string, Node*> enqueue_op; in GetEnqueueOpsFromGraph() local
185 node->type_string(), node->attrs(), node, &enqueue_op, &found_recv_op, in GetEnqueueOpsFromGraph()
192 return ComputeEnqueueTrainingStatus(enqueue_op, found_recv_op, in GetEnqueueOpsFromGraph()
[all …]
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/
H A DClGemmLowpMatrixMultiplyCore.cpp689 CLScheduler::get().enqueue_op(*_mtx_b_reshape_kernel, mtx_b_reshape_pack, false); in run()
701 CLScheduler::get().enqueue_op(*_mtx_b_reduction_kernel, mtx_b_red_pack, false); in run()
712 CLScheduler::get().enqueue_op(*_mtx_a_reduction_kernel, mtx_a_red_pack, false); in run()
742 CLScheduler::get().enqueue_op(*_mm_reshaped_only_rhs_kernel, gemm_reshaped_pack, false); in run()
746 … CLScheduler::get().enqueue_op(*_mm_reshaped_only_rhs_mmul_kernel, gemm_reshaped_pack, false); in run()
761 CLScheduler::get().enqueue_op(*_mm_native_kernel, gemm_native_pack, false); in run()
776 … CLScheduler::get().enqueue_op(*_offset_contribution_output_stage_kernel, output_stage_pack, true); in run()
788 CLScheduler::get().enqueue_op(*_offset_contribution_kernel, offset_contrib_pack, true); in run()
806 CLScheduler::get().enqueue_op(*_weights_to_qasymm8, convert_to_qs8_pack, false); in prepare()
818 CLScheduler::get().enqueue_op(*_mtx_b_reshape_kernel, mtx_b_pack, false); in prepare()
[all …]
H A DClGemm.cpp685 CLScheduler::get().enqueue_op(*_mm_native_kernel, tensors, true); in run()
692 CLScheduler::get().enqueue_op(*_reshape_lhs_kernel, reshape_lhs_pack, false); in run()
698 CLScheduler::get().enqueue_op(*_reshape_rhs_kernel, reshape_rhs_pack, false); in run()
707 CLScheduler::get().enqueue_op(*_mm_reshaped_kernel, gemm_reshaped_pack, true); in run()
717 CLScheduler::get().enqueue_op(*_reshape_rhs_kernel, reshape_rhs_pack, false); in run()
735 … CLScheduler::get().enqueue_op(*_mm_reshaped_only_rhs_kernel, gemm_reshaped_onlyrhs_pack, true); in run()
745 CLScheduler::get().enqueue_op(*_reshape_rhs_kernel, reshape_rhs_pack, false); in run()
763 …CLScheduler::get().enqueue_op(*_mm_reshaped_only_rhs_mmul_kernel, gemm_reshaped_onlyrhs_pack, true… in run()
790 CLScheduler::get().enqueue_op(*_reshape_rhs_kernel, reshape_rhs_pack, true); in prepare()
H A DClDirectConv2d.cpp123 CLScheduler::get().enqueue_op(*_src_border_handler.get(), tensors, false); in run()
125 CLScheduler::get().enqueue_op(*_direct_conv_kernel.get(), tensors, false); in run()
130 CLScheduler::get().enqueue_op(*_activation_kernel.get(), act_pack, false); in run()
H A DClWinogradConv2d.cpp259 CLScheduler::get().enqueue_op(_border_handler, pack_it, false); in run()
260 CLScheduler::get().enqueue_op(*_input_transform, pack_it, false); in run()
276 CLScheduler::get().enqueue_op(*_output_transform, pack_ot); in run()
293 CLScheduler::get().enqueue_op(*_filter_transform, pack_ft, false); in prepare()
H A DClIndirectConv2d.cpp122 CLScheduler::get().enqueue_op(*_indirect_conv_kernel, indirect_conv2d_pack, true); in run()
138 CLScheduler::get().enqueue_op(*_addr_precalculation_kernel, indirect_buffer_pack, true); in prepare()
H A DClGemmConv2d.cpp564 CLScheduler::get().enqueue_op(*_im2col_kernel, pack, false); in run()
599 CLScheduler::get().enqueue_op(*_col2im_kernel.get(), pack, false); in run()
611 CLScheduler::get().enqueue_op(*_activation_kernel.get(), pack, false); in run()
634 CLScheduler::get().enqueue_op(*_weights_reshape_kernel.get(), pack, true); in prepare()
H A DClConcatenate.cpp234 CLScheduler::get().enqueue_op(*_concat_kernels.at(0), tensors, true); in run()
244 CLScheduler::get().enqueue_op(*k, pack, true); in run()
H A DClSoftmax.cpp172 CLScheduler::get().enqueue_op(*_max_shift_exp_sum_kernel.get(), sum_pack, false); in run()
173 CLScheduler::get().enqueue_op(*_norm_kernel.get(), norm_pack, false); in run()
/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
80 enqueue_ops.push_back(enqueue_op); in EnqueueOpsFanin()
133 for (const string& enqueue_op : queue_runner.enqueue_op_name()) { in NodesToPreserve() local
134 result.insert(NodeName(enqueue_op)); in NodesToPreserve()
/aosp_15_r20/external/tensorflow/tensorflow/cc/training/
H A Dqueue_runner.cc98 for (const string& enqueue_op : enqueue_op_names_) { in Start() local
100 std::bind(&QueueRunner::Run, this, sess, enqueue_op)); in Start()
162 void QueueRunner::Run(Session* sess, const string& enqueue_op) { in Run() argument
169 status = RealRun(sess, enqueue_op, true); in Run()
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dqueue_runner_impl.py236 def _run(self, sess, enqueue_op, coord=None): argument
249 enqueue_callable = sess.make_callable(enqueue_op)
370 for enqueue_op in self.enqueue_ops:
372 ops.strip_name_scope(enqueue_op.name, export_scope))
H A Dqueue_runner_test.py317 enqueue_op = control_flow_ops.no_op(name="enqueue")
321 queue, [enqueue_op],
330 self.assertEqual(enqueue_op, qr0_recon.enqueue_ops[0])
343 self.assertEqual(enqueue_op, qr0_legacy_recon.enqueue_ops[0])
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu_embedding_v2.py1274 enqueue_op = self._generate_enqueue_op(
1280 _add_key_attr(enqueue_op, name)
1284 ops.get_default_graph().control_outputs.append(enqueue_op)
1307 enqueue_op = self._generate_enqueue_op(
1313 _add_key_attr(enqueue_op, name)
1314 enqueue_ops.append(enqueue_op)
1324 enqueue_op = self._generate_enqueue_op(
1331 _add_key_attr(enqueue_op, name)
1332 ops.get_default_graph().control_outputs.append(enqueue_op)
/aosp_15_r20/external/ComputeLibrary/tests/CL/
H A DHelper.h114 CLScheduler::get().enqueue_op(_border_handler, tensors); in run()
115 CLScheduler::get().enqueue_op(*_kernel, tensors); in run()
247 CLScheduler::get().enqueue_op(*_kernel, tensors); in run()
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLQLSTMLayer.cpp1140 … CLScheduler::get().enqueue_op(*_input_to_input_reduction, input_to_input_red_pack, false); in prepare()
1143 … CLScheduler::get().enqueue_op(*_recurrent_to_input_reduction, rec_to_input_red_pack, false); in prepare()
1160 CLScheduler::get().enqueue_op(*_input_to_forget_reduction, input_to_forget_red_pack, false); in prepare()
1163 … CLScheduler::get().enqueue_op(*_recurrent_to_forget_reduction, rec_to_forget_red_pack, false); in prepare()
1166 CLScheduler::get().enqueue_op(*_input_to_cell_reduction, input_to_cell_red_pack, false); in prepare()
1169 CLScheduler::get().enqueue_op(*_recurrent_to_cell_reduction, rec_to_cell_red_pack, false); in prepare()
1172 CLScheduler::get().enqueue_op(*_input_to_output_reduction, input_to_output_red_pack, false); in prepare()
1175 … CLScheduler::get().enqueue_op(*_recurrent_to_output_reduction, rec_to_output_red_pack, false); in prepare()
1181 CLScheduler::get().enqueue_op(*_projection_reduction, proj_red_pack, false); in prepare()
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/runtime/gpu/cl/
H A DClWorkloadRuntime.cpp333 … CLScheduler::get().enqueue_op(*kernel, _impl->_tensor_lut.get_tensor_pack(uwk_id), flush_queue); in prepare()
353 … CLScheduler::get().enqueue_op(*kernel, _impl->_tensor_lut.get_tensor_pack(uwk_id), flush_queue); in run()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dexhaustive_op_test_utils.h184 void Run(EnqueueOp enqueue_op, EvaluateOp evaluate_op) {
185 Run(enqueue_op, evaluate_op, GetDefaultSpecGenerator<T, N>());
195 void Run(EnqueueOp enqueue_op, EvaluateOp evaluate_op,
206 EnqueueOpWrapper<XlaInputs, N>::BuildFromInputs(xla_inputs, enqueue_op);
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_experimental.cc296 TF_Operation* enqueue_op = TF_GraphOperationByName( in TF_EnqueueNamedTensor() local
299 if (enqueue_op == nullptr) { in TF_EnqueueNamedTensor()
322 /*targets*/ &enqueue_op, /*ntargets*/ 1, in TF_EnqueueNamedTensor()
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/
H A DCLOperator.cpp45 CLScheduler::get().enqueue_op(*_kernel.get(), tensors, false); in run()

12