Home
last modified time | relevance | path

Searched defs:output_shape_ptr (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dtile_gpu_kernel.h44 size_t *output_shape_ptr = GetDeviceAddress<size_t>(workspace, 1); in Launch() local
/third_party/mindspore/mindspore/core/abstract/
Dprim_nn.cc177 ShapePtr output_shape_ptr = std::make_shared<Shape>(x_shape, x_min_shape, x_max_shape); in InferImplBatchNorm() local
340 …ShapePtr output_shape_ptr = std::make_shared<Shape>(output_shape, output_shape_min, output_shape_m… in InferImplConv2D() local