Searched defs:input_ref (Results 1 – 7 of 7) sorted by relevance
237 TensorReference input_ref(*input); in DeviceToDeviceCopy() local287 TensorReference input_ref(*gpu_tensor); in CopyGPUTensorToCPU() local333 TensorReference input_ref(*cpu_tensor); in CopyCPUTensorToGPU() local
154 TensorRef<Tensor<float, 3>> input_ref(input); in test_ref_in_expr() local
308 const T& input_ref = tensor_in_mat.coeffRef(d, in_index); in Compute() local
149 const T& input_ref = in_mat.coeffRef(d, in_index); in SpatialMaxPoolWithArgMaxHelper() local623 const T& input_ref = in_mat.coeffRef(d, in_index); in SpatialMaxPoolGradGrad() local
631 const T& input_ref = in_mat.coeffRef(d, in_index); in launch() local
16036 std::string input_ref; in prepareDrawShaderSource() local
6871 VkAttachmentReference input_ref = {0, VK_IMAGE_LAYOUT_GENERAL}; in TEST_F() local