Searched refs:input_ref (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 237 TensorReference input_ref(*input); in DeviceToDeviceCopy() local
240 [done, send_device_to_device_stream, input_ref]() { in DeviceToDeviceCopy()
241 input_ref.Unref(); in DeviceToDeviceCopy()
287 TensorReference input_ref(*gpu_tensor); in CopyGPUTensorToCPU() local
290 [send_device_to_host_stream, done, input_ref]() { in CopyGPUTensorToCPU()
294 input_ref.Unref(); in CopyGPUTensorToCPU()
333 TensorReference input_ref(*cpu_tensor); in CopyCPUTensorToGPU() local
336 [recv_host_to_device_stream, done, input_ref]() { in CopyCPUTensorToGPU()
337 input_ref.Unref(); in CopyCPUTensorToGPU()
|
| /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 | 149 const T& input_ref = in_mat.coeffRef(d, in_index); in SpatialMaxPoolWithArgMaxHelper() local
152 if (output_ref < input_ref || in SpatialMaxPoolWithArgMaxHelper()
154 output_ref = input_ref; in SpatialMaxPoolWithArgMaxHelper()
623 const T& input_ref = in_mat.coeffRef(d, in_index); in SpatialMaxPoolGradGrad() local
624 if (output_ref == input_ref) { in SpatialMaxPoolGradGrad()
|
| D | pooling_ops_3d.cc | 631 const T& input_ref = in_mat.coeffRef(d, in_index); in launch() local
632 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 | layer_validation_tests.cpp | 6871 VkAttachmentReference input_ref = {0, VK_IMAGE_LAYOUT_GENERAL}; in TEST_F() local
6874 …VkSubpassDescription subpass = {0, VK_PIPELINE_BIND_POINT_GRAPHICS, 1, &input_ref, 1, &color_ref, … in TEST_F()
|