Searched refs:output_shape_ptr (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | tile_gpu_kernel.h | 44 size_t *output_shape_ptr = GetDeviceAddress<size_t>(workspace, 1); in Launch() local
53 cudaMemcpyAsync(output_shape_ptr, &output_shape_[0], output_shape_.size() * sizeof(size_t), in Launch()
56 … CalTile(output_size_, input_size_, shape_size_, input_shape_ptr, output_shape_ptr, input, output, in Launch()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_nn.cc | 177 ShapePtr output_shape_ptr = std::make_shared<Shape>(x_shape, x_min_shape, x_max_shape); in InferImplBatchNorm() local
178 AbstractTensorPtr output = std::make_shared<AbstractTensor>(input_x->element(), output_shape_ptr); in InferImplBatchNorm()
340 …ShapePtr output_shape_ptr = std::make_shared<Shape>(output_shape, output_shape_min, output_shape_m… in InferImplConv2D() local
341 return std::make_shared<AbstractTensor>(x_type, output_shape_ptr); in InferImplConv2D()
|