Searched refs:BROADCAST_TO (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | broadcast_to_fp32.cc | 62 return BROADCAST_TO(float, reinterpret_cast<const float *>(input_data), &shape_info_, in Run()
66 return BROADCAST_TO(float16_t, reinterpret_cast<const float16_t *>(input_data), &shape_info_, in Run()
71 return BROADCAST_TO(int, reinterpret_cast<const int *>(input_data), &shape_info_, in Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | broadcast_to_cpu_kernel.cc | 70 status = BROADCAST_TO(bool, input_addr, &shape_info_, output_addr); in Launch()
72 status = BROADCAST_TO(int, input_addr, &shape_info_, output_addr); in Launch()
74 status = BROADCAST_TO(float, input_addr, &shape_info_, output_addr); in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | broadcast_to.h | 24 #define BROADCAST_TO(type, input, shape_info, output) broadcast_to_##type(input, shape_info, output) macro
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | broadcast_to_info.cc | 165 …auto new_broadcast_to = gen_g.PushBack({gen_g.NewOpInst(BROADCAST_TO, attrs), gen_g.virtual_input_… in ComputeReplaceGraph()
|
| D | ops_utils.h | 333 constexpr char BROADCAST_TO[] = "BroadcastTo"; variable
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_auto_parallel.cc | 169 …EMBEDDING_LOOKUP, FUSE_BATCH_NORM_EX, SPLIT, BROADCAST_TO, ABS, ACOSH, ASIN, ASINH, ATAN, ATANH, C… in IsSplittableOperator()
|