Home
last modified time | relevance | path

Searched defs:sample_shape (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/random/
Drandom_gamma_grad_gpu_kernel.cc74 std::vector<int64_t> sample_shape = inputs.at(kIndex1)->GetShapeVector(); in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/
Drandomgammagrad_helper.h50 auto sample_shape = input_shapes[1]; in CalMemSize() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Drandom_gamma_grad.cc55 auto sample_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(sample_shape_ptr)[kShape]; in RandomGammaGradInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Drandom_gamma_grad_impl.cu193 …tRandomGammaGrad(const std::vector<int64_t> &alpha_shape, const std::vector<int64_t> &sample_shape, in BroadcastRandomGammaGrad()