Searched defs:input_reshape (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | bce_with_logits_loss_kernel.cc | 184 std::vector<size_t> input_reshape = {input_size_}; in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/ |
| D | expand_cpu_kernel.cc | 154 Eigen::DSizes<Eigen::DenseIndex, RANK> input_reshape; in ExpandCalculate() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | reduce_impl.cu | 756 void ReduceImpl(IN_T input, const std::vector<size_t> &input_reshape, const bool reduce_first_axis,… in ReduceImpl()
784 cudaError_t ArrayReduce(T *input, const std::vector<size_t> &input_reshape, const bool reduce_first… in ArrayReduce()
832 cudaError_t ArrayReduce(half *input, const std::vector<size_t> &input_reshape, const bool reduce_fi… in ArrayReduce()
890 cudaError_t ArrayReduceComplex(T *input, const std::vector<size_t> &input_reshape, const bool reduc… in ArrayReduceComplex()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | encoder_layer_fusion.cc | 309 VectorRef EncoderLayerFusion::DefinePatternMoEFfn(VectorRef input_reshape, bool gelu = false) const… in DefinePatternMoEFfn()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/audio/kernels/ |
| D | audio_utils.cc | 1961 …TensorShape input_reshape({input->Size() / input_shape[-1] / input_shape[-2] / input_shape[-3], in… in InverseSpectrogramImpl() local
2166 …TensorShape input_reshape({input->Size() / input_shape[-1] / input_shape[-2], input_shape[-2], inp… in InverseMelScaleImpl() local
|