Searched refs:output_elem (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parallel_interleave_dataset_op.cc | 633 OutputElem output_elem; member
646 WorkerThreadState() : output_elem(Status::OK()) {} in WorkerThreadState()
827 if (worker_thread_states_[thread_index].output_elem.status.ok() && in WorkerThread()
829 .output_elem.output.empty() && in WorkerThread()
831 int64& id = worker_thread_states_[thread_index].output_elem.id; in WorkerThread()
839 worker_thread_states_[thread_index].output_elem.status = in WorkerThread()
842 &worker_thread_states_[thread_index].output_elem.output, in WorkerThread()
882 worker_thread_states_[thread_index].output_elem.status, in WorkerThread()
883 worker_thread_states_[thread_index].output_elem.id); in WorkerThread()
885 worker_thread_states_[thread_index].output_elem.output); in WorkerThread()
[all …]
|
| /external/tensorflow/tensorflow/cc/tools/ |
| D | freeze_saved_model.cc | 63 for (auto& output_elem : signature_def.outputs()) { in GetSignatureDefsInputsAndOutputs() local
64 GetTensorNamesFromTensorInfo(output_elem.second, outputs); in GetSignatureDefsInputsAndOutputs()
|
| /external/mesa3d/src/amd/vulkan/ |
| D | radv_query.c | 449 nir_ssa_def *output_elem = nir_iadd(&b, output_base, in build_pipeline_statistics_query_shader() local
456 store->src[2] = nir_src_for_ssa(output_elem); in build_pipeline_statistics_query_shader()
467 store->src[2] = nir_src_for_ssa(output_elem); in build_pipeline_statistics_query_shader()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | padding_fifo_queue_test.py | 1591 for (input_elem, output_elem) in zip(input_tuple, output_tuple):
1592 self.assertAllEqual(input_elem, output_elem)
|
| D | fifo_queue_test.py | 614 for (input_elem, output_elem) in zip(input_tuple, output_tuple):
615 self.assertAllEqual(input_elem, output_elem)
|