Searched full:batch_descriptor (Results 1 – 11 of 11) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 614 CudnnTensorDescriptor(const dnn::BatchDescriptor& batch_descriptor, in CudnnTensorDescriptor() argument
617 switch (batch_descriptor.layout()) { in CudnnTensorDescriptor()
620 const int nd = batch_descriptor.ndims() + 2; in CudnnTensorDescriptor()
623 batch_descriptor.full_strides(dnn::DataLayout::kBatchDepthYX); in CudnnTensorDescriptor()
625 batch_descriptor.full_dims(dnn::DataLayout::kBatchDepthYX); in CudnnTensorDescriptor()
636 << "batch_descriptor: " << batch_descriptor.ToString(); in CudnnTensorDescriptor()
642 batch_descriptor.layout() == dnn::DataLayout::kBatchDepthYX4 in CudnnTensorDescriptor()
648 batch_descriptor.count(), batch_descriptor.feature_map_count(), in CudnnTensorDescriptor()
649 batch_descriptor.height(), batch_descriptor.width())) in CudnnTensorDescriptor()
650 << "batch_descriptor: " << batch_descriptor.ToString(); in CudnnTensorDescriptor()
[all …]
|
| H A D | cuda_dnn.h | 390 Stream* stream, const dnn::BatchDescriptor& batch_descriptor, in DoSeparableConvolve() argument
504 const dnn::BatchDescriptor& batch_descriptor,
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 651 ScopedTensorDescriptor(const BatchDescriptor& batch_descriptor, in ScopedTensorDescriptor() argument
660 switch (batch_descriptor.layout()) { in ScopedTensorDescriptor()
663 const int nd = batch_descriptor.ndims() + 2; in ScopedTensorDescriptor()
667 batch_descriptor.full_strides(dnn::DataLayout::kBatchDepthYX); in ScopedTensorDescriptor()
669 batch_descriptor.full_dims(dnn::DataLayout::kBatchDepthYX); in ScopedTensorDescriptor()
683 << batch_descriptor.ToString() in ScopedTensorDescriptor()
689 << DataLayoutString(batch_descriptor.layout()); in ScopedTensorDescriptor()
4524 const BatchDescriptor& batch_descriptor, in DeriveOutputBatchDescriptor() argument
4528 ScopedTensorDescriptor input_nd{batch_descriptor, miopenFloat}; in DeriveOutputBatchDescriptor()
4532 int dn = batch_descriptor.ndims() + 2; in DeriveOutputBatchDescriptor()
[all …]
|
| H A D | rocm_dnn.h | 394 Stream* stream, const dnn::BatchDescriptor& batch_descriptor, in DoSeparableConvolve() argument
510 const dnn::BatchDescriptor& batch_descriptor,
|
| /aosp_15_r20/external/googleapis/google/pubsub/v1/ |
| H A D | pubsub_gapic.yaml | 20 batch_descriptor:
|
| /aosp_15_r20/external/googleapis/google/logging/v2/ |
| H A D | logging_gapic.yaml | 23 batch_descriptor:
|
| /aosp_15_r20/external/sdk-platform-java/gapic-generator-java/src/test/resources/ |
| H A D | logging_gapic.yaml | 74 batch_descriptor:
|
| H A D | pubsub_gapic.yaml | 244 batch_descriptor:
|
| /aosp_15_r20/external/sdk-platform-java/gapic-generator-java/src/main/java/com/google/api/generator/gapic/protoparser/ |
| H A D | BatchingSettingsConfigParser.java | 43 private static String YAML_KEY_DESCRIPTOR = "batch_descriptor";
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.h | 1322 const BatchDescriptor& batch_descriptor, in PrepareForConvolution() argument
1333 kind, ToDataType<ElementType>::value, stream, batch_descriptor, in PrepareForConvolution()
2621 const BatchDescriptor& batch_descriptor, DeviceMemoryBase input_data, in DoPrepareForConvolution() argument
|
| H A D | stream.cc | 512 const dnn::BatchDescriptor &batch_descriptor, in ThenSeparableConvolve() argument
521 PARAM(batch_descriptor), PARAM(input_data), PARAM(filter_descriptor), in ThenSeparableConvolve()
527 this, batch_descriptor, input_data, filter_descriptor, depth_multiplier, in ThenSeparableConvolve()
|