| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | BUILD | 71 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
110 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
116 name = "tensor_desc",
117 srcs = ["tensor_desc.cc"],
118 hdrs = ["tensor_desc.h"],
151 ":tensor_desc",
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | BUILD | 49 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
101 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
161 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
175 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | BUILD | 46 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
310 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
418 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
442 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
517 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
537 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
548 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
| D | serialization.cc | 663 const auto* tensor_desc = in Encode() local
665 if (!tensor_desc) continue; in Encode()
666 auto desc_fb = Encode(*tensor_desc, builder); in Encode()
676 const auto* tensor_desc = in Encode() local
678 if (!tensor_desc) continue; in Encode()
679 auto desc_fb = Encode(*tensor_desc, builder); in Encode()
716 const auto* tensor_desc = in Encode() local
718 if (!tensor_desc) continue; in Encode()
719 auto desc_fb = Encode(*tensor_desc, builder); in Encode()
729 const auto* tensor_desc = in Encode() local
[all …]
|
| D | cl_arguments.cc | 195 const auto* tensor_desc = dynamic_cast<const TensorDescriptor*>(desc); in CreateCLObject() local
196 if (tensor_desc) { in CreateCLObject()
198 RETURN_IF_ERROR(gpu_tensor.CreateFromDescriptor(*tensor_desc, context)); in CreateCLObject()
351 const auto* tensor_desc = dynamic_cast<const TensorDescriptor*>(desc_ptr); in ResolveSelector() local
352 if (tensor_desc && (selector == "Write" || selector == "Linking")) { in ResolveSelector()
361 RETURN_IF_ERROR(tensor_desc->GetLinkingContextFromWriteSelector( in ResolveSelector()
|
| D | tensor.cc | 347 const auto* tensor_desc = dynamic_cast<const TensorDescriptor*>(obj_ptr); in GetGPUResources() local
348 if (!tensor_desc) { in GetGPUResources()
352 {"slice_stride", tensor_desc->GetSliceStrideSize(shape_)}); in GetGPUResources()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | BUILD | 106 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
148 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
214 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
250 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
271 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
292 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
326 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
345 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
379 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
469 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/ |
| D | BUILD | 16 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
35 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | metal_arguments.cc | 152 const auto* tensor_desc = dynamic_cast<const TensorDescriptor*>(desc); in CreateMetalObject() local
153 if (tensor_desc) { in CreateMetalObject()
155 RETURN_IF_ERROR(gpu_tensor.CreateFromDescriptor(*tensor_desc, device)); in CreateMetalObject()
734 const auto* tensor_desc = dynamic_cast<const TensorDescriptor*>(desc_ptr); in ResolveSelector() local
735 if (tensor_desc && (selector == "Write" || selector == "Linking")) { in ResolveSelector()
744 RETURN_IF_ERROR(tensor_desc->GetLinkingContextFromWriteSelector( in ResolveSelector()
|
| D | BUILD | 190 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
247 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
| D | metal_spatial_tensor.cc | 252 const auto* tensor_desc = dynamic_cast<const TensorDescriptor*>(obj_ptr); in GetGPUResources() local
253 if (!tensor_desc) { in GetGPUResources()
257 {"slice_stride", tensor_desc->GetSliceStrideSize(shape_)}); in GetGPUResources()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1482 TensorDescriptor tensor_desc = CreateTensorDescriptor(); in Create() local
1484 /*tensorDesc=*/tensor_desc.get(), /*dataType=*/data_type, in Create()
1490 std::move(tensor_desc)); in Create()
1502 TensorDescriptor tensor_desc = CreateTensorDescriptor(); in Create() local
1504 /*tensorDesc=*/tensor_desc.get(), /*dataType=*/data_type, in Create()
1525 std::move(data_desc), std::move(tensor_desc)); in Create()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | BUILD | 146 "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 380 uint64 GetHashValue(miopenTensorDescriptor_t tensor_desc) { in GetHashValue() argument
384 wrap::miopenGetTensorDescriptor(tensor_desc, &datatype, dims, strides); in GetHashValue()
|