| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_chipping_sycl.cpp | 50 TensorMap<Tensor<DataType, 5, DataLayout,IndexType>> gpu_tensor(gpu_data_tensor, tensorRange); in test_static_chip_sycl() local
199 TensorMap<Tensor<DataType, 5, DataLayout,IndexType>> gpu_tensor(gpu_data_tensor, tensorRange); in test_dynamic_chip_sycl() local
350 TensorMap<Tensor<DataType, 5, DataLayout,IndexType>> gpu_tensor(gpu_data_tensor, tensorRange); in test_chip_in_expr() local
427 TensorMap<Tensor<DataType, 5, DataLayout,IndexType>> gpu_tensor(gpu_data_tensor, tensorRange); in test_chip_as_lvalue_sycl() local
|
| H A D | cxx11_tensor_striding_sycl.cpp | 53 …Eigen::TensorMap<Eigen::Tensor<DataType, 4, DataLayout, IndexType> > gpu_tensor(d_tensor, tensor_d… in test_simple_striding() local
136 …Eigen::TensorMap<Eigen::Tensor<DataType, 4, DataLayout, IndexType> > gpu_tensor(d_tensor, tensor_d… in test_striding_as_lvalue() local
|
| H A D | cxx11_tensor_inflation_sycl.cpp | 56 TensorMap<Tensor<DataType, 4, DataLayout,IndexType>> gpu_tensor(gpu_data_tensor, tensorRange); in test_simple_inflation_sycl() local
|
| H A D | cxx11_tensor_patch_sycl.cpp | 58 TensorMap<Tensor<DataType, 4, DataLayout,IndexType>> gpu_tensor(gpu_data_tensor, tensorRange); in test_simple_patch_sycl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/ |
| H A D | conversion_function.cc | 46 const tfrt::gpu::DenseGpuTensor& gpu_tensor, Device* device, in CopyRefGpuTensorToRuntimeFallbackTensor()
146 tfrt::gpu::DenseGpuTensor gpu_tensor{ in ConvertRuntimeFallbackTensorToDenseGpuTensor() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_util.cc | 268 const Tensor* gpu_tensor, Tensor* cpu_tensor, in CopyGPUTensorToCPU()
313 Device* gpu_device, Tensor* gpu_tensor, in CopyCPUTensorToGPU()
|
| H A D | gpu_device_test.cc | 114 void CopyCPUToGPU(Tensor* cpu_tensor, Tensor* gpu_tensor, Device* device, in CopyCPUToGPU()
120 void CopyGPUToCPU(Tensor* gpu_tensor, Tensor* cpu_tensor, Device* device, in CopyGPUToCPU()
|
| H A D | gpu_device.cc | 589 Tensor BaseGPUDevice::CopyGpuTensorToHostDebugOnly(const Tensor& gpu_tensor) { in CopyGpuTensorToHostDebugOnly()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/ |
| H A D | op_handler_selector_test.cc | 403 tensorflow::core::RefCountPtr<FakeTensorHandle> gpu_tensor( in TEST_F() local
432 tensorflow::core::RefCountPtr<FakeTensorHandle> gpu_tensor( in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| H A D | inference_context.cc | 347 MetalSpatialTensor* gpu_tensor = GetTensor(id); in SetInputTensor() local
355 const MetalSpatialTensor* gpu_tensor = GetTensor(id); in GetOutputTensor() local
|
| H A D | metal_arguments.cc | 76 MetalSpatialTensor gpu_tensor; in CreateMetalObject() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| H A D | trt_engine_op.cc | 549 const Tensor& gpu_tensor = ctx->input(i); in CopyToHostAsync() local
563 Tensor* gpu_tensor; in CopyToDeviceAsync() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| H A D | inference_context.cc | 905 Tensor* gpu_tensor = GetTensor(id); in SetInputTensor() local
914 const Tensor* gpu_tensor = GetTensor(id); in GetOutputTensor() local
|
| H A D | cl_arguments.cc | 115 Tensor gpu_tensor; in CreateCLObject() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | direct_session_test.cc | 2401 Tensor gpu_tensor; in TEST() local
2474 Tensor gpu_tensor; in TestFeedAndFetchTensorsInDeviceMemory() local
|