| /third_party/mindspore/mindspore/_extends/parallel_compile/tbe_compiler/ |
| D | compiler.py | 97 is_dynamic_shape = kernel_info['op_info']['is_dynamic_shape']
98 if is_dynamic_shape:
105 if is_dynamic_shape:
120 if is_dynamic_shape:
|
| D | helper.py | 85 is_dynamic_shape = kernel_info['op_info']['is_dynamic_shape']
101 elif is_dynamic_shape:
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | layer_norm_grad_split.cc | 109 bool is_dynamic_shape = AnfAlgo::IsDynamicShape(cnode); in Process() local
112 …reateOutputsOfLayerNormXBackpropV2(graph, cnode, &layer_norm_x_backprop_outputs, is_dynamic_shape); in Process()
121 … &layer_norm_beta_gamma_backprop_outputs, is_dynamic_shape); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | pynative_task_manager.h | 33 …RunOpContext(std::string graph_info, bool is_dynamic_shape, KernelGraphPtr graph, std::vector<int6… in RunOpContext() argument
37 is_dynamic_shape_(is_dynamic_shape), in RunOpContext()
45 bool is_dynamic_shape() const { return is_dynamic_shape_; } in is_dynamic_shape() function
|
| D | ascend_session.cc | 332 if (!kernel_graph->is_dynamic_shape() && IsVMGraphTaskSink() && in TensorNeedSync()
835 … bool is_dynamic_shape, const std::vector<tensor::TensorPtr> &input_tensors) { in LaunchFunc() argument
851 if (is_dynamic_shape) { in LaunchFunc()
939 … op_run_info.is_dynamic_shape || ms_context->get_param<bool>(MS_CTX_ENABLE_PYNATIVE_SYNCHRONIZE); in DisableLazyBuild()
963 LaunchFunc(graph, tensor_to_node, op_run_info->is_dynamic_shape, *input_tensors); in RunOpImpl()
965 …auto run_op_context = std::make_shared<RunOpContext>(graph_info, op_run_info->is_dynamic_shape, gr… in RunOpImpl()
1006 if (op_run_info->is_dynamic_shape) { in RunOpImplOrigin()
1017 if (op_run_info->is_dynamic_shape) { in RunOpImplOrigin()
1136 if (op_run_info.is_dynamic_shape) { in BuildOpsInGraph()
1994 …LaunchFunc(context->graph(), context->tensor_to_node(), context->is_dynamic_shape(), context->inpu… in ExecuteAllTaskInQueue()
|
| D | ascend_session.h | 161 …const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, bool is_dynamic_shape,
|
| D | cpu_session.cc | 306 if (op_run_info->is_dynamic_shape) { in RunOpImpl()
|
| D | session_basic.h | 63 bool is_dynamic_shape = false; member
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_dynaminc_shape_util.cc | 350 auto is_dynamic_shape = AnfAlgo::HasNodeAttr(kAttrIsDynamicShape, cnode); in GetDynamicShapeAttr() local
351 if (!is_dynamic_shape) { in GetDynamicShapeAttr()
354 is_dynamic_shape = AnfAlgo::GetNodeAttr<bool>(cnode, kAttrIsDynamicShape); in GetDynamicShapeAttr()
355 return is_dynamic_shape; in GetDynamicShapeAttr()
370 auto is_dynamic_shape = GetDynamicShapeAttr(cnode); in FindOp() local
371 return mindspore::kernel::OpLib::FindOp(op_name, OpImplyType::kTBE, is_dynamic_shape); in FindOp()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/executor/ |
| D | dynamic_kernel.h | 43 [[nodiscard]] bool is_dynamic_shape() const { return is_dynamic_shape_; } in is_dynamic_shape() function
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | base.h | 53 bool is_dynamic_shape = false; member
|
| D | pynative_cache.h | 64 bool is_dynamic_shape = false; member
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/oplib/ |
| D | oplib.h | 35 bool is_dynamic_shape = false);
|
| D | oplib.cc | 353 …<OpInfo> OpLib::FindOp(const std::string &op_name, OpImplyType imply_type, bool is_dynamic_shape) { in FindOp() argument
375 if (is_dynamic_shape && !op_info->dynamic_shape()) { in FindOp()
381 … << ", current op num: " << op_info_.size() << " is_dynamic_shape:" << is_dynamic_shape; in FindOp()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/ |
| D | cpu_device_context.h | 58 bool is_dynamic_shape = false) const override;
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.cc | 276 bool is_dynamic_shape = false; in NewTransOpNode() local
284 is_dynamic_shape = true; in NewTransOpNode()
294 if (is_dynamic_shape) { in NewTransOpNode()
313 if (is_dynamic_shape) { in NewTransOpNode()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | constant_folding_fusion.cc | 71 bool is_dynamic_shape = false; in GetCNodeInputTensors() local
75 is_dynamic_shape = true; in GetCNodeInputTensors()
79 if (!is_dynamic_shape) { in GetCNodeInputTensors()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/ |
| D | device_context.h | 103 bool is_dynamic_shape = false) const = 0;
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/ |
| D | gpu_device_context.h | 64 bool is_dynamic_shape = false) const override;
|
| D | gpu_device_context.cc | 393 bool is_dynamic_shape) const { in LaunchKernel()
431 if (is_dynamic_shape) { in LaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 482 if (graph.is_dynamic_shape()) { in Run()
506 if (!is_first_step_map_[graph_id] || graph->is_dynamic_shape()) { in RunOneStep()
768 if (dynamic_kernel && dynamic_kernel->is_dynamic_shape()) { in LaunchKernelDynamic()
795 … if (gpu_kernel != nullptr && dynamic_kernel != nullptr && dynamic_kernel->is_dynamic_shape()) { in LaunchKernelDynamic()
888 if (dynamic_kernel && dynamic_kernel->is_dynamic_shape()) { in RunOpLaunchKernelDynamic()
903 if (gpu_kernel != nullptr && dynamic_kernel != nullptr && dynamic_kernel->is_dynamic_shape()) { in RunOpLaunchKernelDynamic()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_kernel_runtime.cc | 452 if (graph.is_dynamic_shape()) { in GenTask()
515 if (graph.is_dynamic_shape()) { in LoadTask()
903 if (dynamic_kernel->is_dynamic_shape()) { in RunDynamicKernelAsync()
923 if (graph.is_dynamic_shape()) { in RunTask()
|
| D | kernel_build_ascend.cc | 421 bool is_dynamic_graph = kernel_graph->is_dynamic_shape(); in KernelBuildPreprocess()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | graph_partition.cc | 515 void AddNode(const AnfNodePtr &node, bool is_dynamic_shape) { in AddNode()
516 if (is_dynamic_shape) { in AddNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | helper.cc | 328 bool is_dynamic_graph = graph->is_dynamic_shape(); in HideNopNode()
351 bool is_dynamic_graph = graph->is_dynamic_shape(); in RemoveNopNode()
|