Home
last modified time | relevance | path

Searched refs:output_elem (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparallel_interleave_dataset_op.cc565 OutputElem output_elem; member
578 WorkerThreadState() : output_elem(Status::OK()) {} in WorkerThreadState()
750 .output_elem.status.ok() && in WorkerThread()
752 .output_elem.output.empty() && in WorkerThread()
754 worker_thread_states_[thread_index].output_elem.status = in WorkerThread()
758 .output_elem.output, in WorkerThread()
798 worker_thread_states_[thread_index].output_elem.status); in WorkerThread()
800 worker_thread_states_[thread_index].output_elem.output); in WorkerThread()
802 worker_thread_states_[thread_index].output_elem.status = in WorkerThread()
902 writer, worker_thread_states_[index].output_elem, in WriteWorkerThreadStateLocked()
[all …]
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model.cc59 for (auto& output_elem : signature_def.outputs()) { in GetSignatureDefsInputsAndOutputs() local
60 GetTensorNamesFromTensorInfo(output_elem.second, outputs); in GetSignatureDefsInputsAndOutputs()
/external/mesa3d/src/amd/vulkan/
Dradv_query.c485 nir_ssa_def *output_elem = nir_iadd(&b, output_base, in build_pipeline_statistics_query_shader() local
497 store->src[2] = nir_src_for_ssa(output_elem); in build_pipeline_statistics_query_shader()
507 store->src[2] = nir_src_for_ssa(output_elem); in build_pipeline_statistics_query_shader()
/external/tensorflow/tensorflow/python/kernel_tests/
Dpadding_fifo_queue_test.py1525 for (input_elem, output_elem) in zip(input_tuple, output_tuple):
1526 self.assertAllEqual(input_elem, output_elem)
Dfifo_queue_test.py1411 for (input_elem, output_elem) in zip(input_tuple, output_tuple):
1412 self.assertAllEqual(input_elem, output_elem)