Searched defs:output_dimensions (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.h | 412 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
422 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
|
| D | cuda_dnn.cc | 5958 const dnn::BatchDescriptor& output_dimensions, in DoMatMul()
6225 const dnn::BatchDescriptor& output_dimensions, DeviceMemoryBase output_data, in DoPoolForward()
6291 const dnn::BatchDescriptor& output_dimensions, DeviceMemoryBase output_data, in DoPoolBackward()
6449 dnn::BatchDescriptor output_dimensions = in DoDepthConcatenate() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 416 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
426 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
|
| D | rocm_dnn.cc | 3829 const dnn::BatchDescriptor& output_dimensions, in DoMatMul()
4022 const dnn::BatchDescriptor& output_dimensions, DeviceMemoryBase output_data, in DoPoolForward()
4091 const dnn::BatchDescriptor& output_dimensions, in IsSame()
4106 const dnn::BatchDescriptor& output_dimensions, in find()
4124 const dnn::BatchDescriptor& output_dimensions, in insert()
4177 const dnn::BatchDescriptor& output_dimensions, DeviceMemoryBase output_data, in DoPoolBackward()
4438 dnn::BatchDescriptor output_dimensions = in DoDepthConcatenate() local
4482 const dnn::BatchDescriptor& output_dimensions, in DoElementwiseOperate()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | sparse_to_dense.cc | 42 const int output_dimensions = NumElements(output_shape); in Resize() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream.cc | 539 const dnn::BatchDescriptor &output_dimensions, in ThenMatMul()
557 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized()
577 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized()
747 const dnn::BatchDescriptor &output_dimensions, in ThenReshape()
803 const dnn::BatchDescriptor &output_dimensions, in ThenElementwiseOperate()
824 const dnn::BatchDescriptor &output_dimensions, in ThenElementwiseOperateScaledQuantized()
|
| D | dnn.h | 1776 const dnn::BatchDescriptor& output_dimensions, in DoReshape()
1889 const dnn::BatchDescriptor& output_dimensions, in DoElementwiseOperateScaledQuantized()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 171 std::vector<int64_t> output_dimensions(window.dimensions_size()); in InferWindowOutputShape() local
754 std::vector<int64_t> output_dimensions(lhs.rank()); in InferDegenerateDimensionBroadcastShape() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 1377 std::vector<int64_t> output_dimensions(window.dimensions_size()); in InferWindowOutputShape() local
|