Searched refs:range_delta (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | dynamic_range_impl.cu | 23 …l__ void ValidateInputAndInferShape(const T *range_start, const T *range_end, const T *range_delta, in ValidateInputAndInferShape() argument
28 T delta = range_delta[0]; in ValidateInputAndInferShape()
65 __global__ void Range(const T *range_start, const T *range_end, const T *range_delta, T *output, in… in Range() argument
68 T delta = range_delta[0]; in Range()
77 void CudaValidateInputAndInferShape(const T *range_start, const T *range_end, const T *range_delta, in CudaValidateInputAndInferShape() argument
80 …ValidateInputAndInferShape<<<1, 1, 0, cuda_stream>>>(range_start, range_end, range_delta, output_s… in CudaValidateInputAndInferShape()
85 void CalRange(const T *range_start, const T *range_end, const T *range_delta, T *output, int64_t *o… in CalRange() argument
87 …<<<GET_BLOCKS(max_output_size), GET_THREADS, 0, cuda_stream>>>(range_start, range_end, range_delta, in CalRange()
91 …lidateInputAndInferShape<int>(const int *range_start, const int *range_end, const int *range_delta,
95 … const int64_t *range_delta, int64_t *output_shape,
[all …]
|
| D | dynamic_range_impl.cuh | 31 void CudaValidateInputAndInferShape(const T *range_start, const T *range_end, const T *range_delta,
36 void CalRange(const T *range_start, const T *range_end, const T *range_delta, T *output, int64_t *o…
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | dynamic_range_gpu_kernel.h | 44 T *range_delta = GetDeviceAddress<T>(inputs, 2); in Launch() local
51 CudaValidateInputAndInferShape(range_start, range_end, range_delta, output_shape_device_address, in Launch()
72 … CalRange(range_start, range_end, range_delta, output_device_address, output_shape_device_address, in Launch()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_arrays.cc | 1023 …AbstractTensorPtr range_delta = CheckArg<AbstractTensor>(op_name, args_spec_list, range_delta_inde… in InferImplRange() local
1028 …TypePtr range_delta_type = CheckTensorDType(range_delta, supported_types, "range_start input of Ra… in InferImplRange()
|