Searched refs:input_ref (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 236 TensorReference input_ref(*input); in DeviceToDeviceCopy() local
239 [done, send_device_to_device_stream, input_ref]() { in DeviceToDeviceCopy()
240 input_ref.Unref(); in DeviceToDeviceCopy()
286 TensorReference input_ref(*gpu_tensor); in CopyGPUTensorToCPU() local
289 [send_device_to_host_stream, done, input_ref]() { in CopyGPUTensorToCPU()
293 input_ref.Unref(); in CopyGPUTensorToCPU()
334 TensorReference input_ref(*cpu_tensor); in CopyCPUTensorToGPU() local
337 [recv_host_to_device_stream, done, input_ref]() { in CopyCPUTensorToGPU()
338 input_ref.Unref(); in CopyCPUTensorToGPU()
|
| /external/rust/crates/rusqlite/src/ |
| D | session.rs | 217 let input_ref = &input; in invert_strm() localVariable
222 input_ref as *const &mut dyn Read as *mut c_void, in invert_strm()
510 let input_ref = &input; in add_stream() localVariable
515 input_ref as *const &mut dyn Read as *mut c_void, in add_stream()
597 let input_ref = &input; in apply_strm() localVariable
607 input_ref as *const &mut dyn Read as *mut c_void, in apply_strm()
616 input_ref as *const &mut dyn Read as *mut c_void, in apply_strm()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fractional_max_pool_op.cc | 308 const T& input_ref = tensor_in_mat.coeffRef(d, in_index); in Compute() local
312 if (output_ref < input_ref || in Compute()
314 output_ref = input_ref; in Compute()
|
| D | maxpooling_op.cc | 157 const T& input_ref = in_mat.coeffRef(d, in_index); in SpatialMaxPoolWithArgMaxHelper() local
161 if (output_ref < input_ref || in SpatialMaxPoolWithArgMaxHelper()
163 output_ref = input_ref; in SpatialMaxPoolWithArgMaxHelper()
631 const T& input_ref = in_mat.coeffRef(d, in_index); in SpatialMaxPoolGradGrad() local
632 if (output_ref == input_ref) { in SpatialMaxPoolGradGrad()
|
| D | pooling_ops_3d.cc | 627 const T& input_ref = in_mat.coeffRef(d, in_index); in launch() local
628 if (output_ref == input_ref) { in launch()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_ref.cpp | 154 TensorRef<Tensor<float, 3>> input_ref(input); in test_ref_in_expr() local
163 result_ref = input_ref + bias; in test_ref_in_expr()
|
| /external/deqp/external/openglcts/modules/gl/ |
| D | gl4cShadingLanguage420PackTests.cpp | 16036 std::string input_ref; in prepareDrawShaderSource() local
16081 input_block_name, input_decl, input_ref); in prepareDrawShaderSource()
16086 input_ref); in prepareDrawShaderSource()
16143 Utils::replaceAllTokens("INPUT_VARIABLE_NAME", input_ref.c_str(), out_source.m_parts[0].m_code); in prepareDrawShaderSource()
|
| /external/vulkan-validation-layers/tests/ |
| D | vklayertests_descriptor_renderpass_framebuffer.cpp | 2766 VkAttachmentReference input_ref = {0, VK_IMAGE_LAYOUT_GENERAL}; in TEST_F() local
2769 …VkSubpassDescription subpass = {0, VK_PIPELINE_BIND_POINT_GRAPHICS, 1, &input_ref, 1, &color_ref, … in TEST_F()
|