Searched defs:input_dims (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | reduce_grad.c | 20 void ReduceMeanByAxes(const float *input_data, int *input_iter, const int *input_dims, int input_nu… in ReduceMeanByAxes()
59 void ReduceSumByAxes(const float *input, const int *input_dims, float *output, const int *output_di… in ReduceSumByAxes()
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/ |
| D | tensorrt_subgraph.cc | 154 nvinfer1::Dims input_dims = ConvertCudaDims(in_tensor.Shape()); in SetTensorRTNetworkInput() local
327 nvinfer1::Dims input_dims = ConvertCudaDims(tensor.Shape()); in Prepare() local
398 nvinfer1::Dims input_dims = ConvertCudaDims(inputs_[i].Shape()); in ReSize() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_property_checker.cc | 47 int input_dims = 0; in CheckStridedSlice() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | one_hot_gpu_kernel.h | 61 int64_t input_dims = static_cast<int64_t>(input_shape.size()); in Init() local
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/ |
| D | shuffle_tensorrt.cc | 272 int ShuffleTensorRT::InferReshapeDims(nvinfer1::Dims input_dims, nvinfer1::Dims *reshape_dims) { in InferReshapeDims()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | transpose_impl_opt.cu | 96 int input_dims[3] = {input_dims_0, input_dims_1, input_dims_2}; in Swap3DTensorLast2DimKernel_shared() local
|
| /third_party/ffmpeg/libavfilter/dnn/ |
| D | dnn_backend_tf.c | 111 int64_t input_dims[] = {1, input->height, input->width, input->channels}; in allocate_input_tensor() local
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | normalization.py | 59 input_dims='2d', argument
|