/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
H A D | padding_fifo_queue_test.py | 97 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 D | fifo_queue_test.py | 463 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 D | priority_queue_test.py | 85 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 D | random_shuffle_queue_test.py | 52 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 D | fifo_queue_test.py | 30 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 D | update_tpu_embedding_ops_passes.cc | 114 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 D | ClGemmLowpMatrixMultiplyCore.cpp | 689 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 D | ClGemm.cpp | 685 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 D | ClDirectConv2d.cpp | 123 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 D | ClWinogradConv2d.cpp | 259 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 D | ClIndirectConv2d.cpp | 122 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 D | ClGemmConv2d.cpp | 564 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 D | ClConcatenate.cpp | 234 CLScheduler::get().enqueue_op(*_concat_kernels.at(0), tensors, true); in run() 244 CLScheduler::get().enqueue_op(*k, pack, true); in run()
|
H A D | ClSoftmax.cpp | 172 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 D | grappler_item.cc | 79 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 D | queue_runner.cc | 98 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 D | queue_runner_impl.py | 236 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 D | queue_runner_test.py | 317 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 D | tpu_embedding_v2.py | 1274 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 D | Helper.h | 114 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 D | CLQLSTMLayer.cpp | 1140 … 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 D | ClWorkloadRuntime.cpp | 333 … 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 D | exhaustive_op_test_utils.h | 184 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 D | c_api_experimental.cc | 296 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 D | CLOperator.cpp | 45 CLScheduler::get().enqueue_op(*_kernel.get(), tensors, false); in run()
|