Searched defs:input_ref (Results 1 – 8 of 8) sorted by relevance
236 TensorReference input_ref(*input); in DeviceToDeviceCopy() local286 TensorReference input_ref(*gpu_tensor); in CopyGPUTensorToCPU() local334 TensorReference input_ref(*cpu_tensor); in CopyCPUTensorToGPU() local
154 TensorRef<Tensor<float, 3>> input_ref(input); in test_ref_in_expr() local
217 let input_ref = &input; in invert_strm() localVariable510 let input_ref = &input; in add_stream() localVariable597 let input_ref = &input; in apply_strm() localVariable
308 const T& input_ref = tensor_in_mat.coeffRef(d, in_index); in Compute() local
157 const T& input_ref = in_mat.coeffRef(d, in_index); in SpatialMaxPoolWithArgMaxHelper() local631 const T& input_ref = in_mat.coeffRef(d, in_index); in SpatialMaxPoolGradGrad() local
627 const T& input_ref = in_mat.coeffRef(d, in_index); in launch() local
2766 VkAttachmentReference input_ref = {0, VK_IMAGE_LAYOUT_GENERAL}; in TEST_F() local
16036 std::string input_ref; in prepareDrawShaderSource() local