| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_manual_gatherv2.py | 39 param_shape=(8, 8)): argument
64 def __init__(self, strategy, split_tuple, param_shape): argument
|
| D | test_gather_field_split.py | 27 def __init__(self, strategy1, strategy2, split_tuple, param_shape, mul_weight_shape): argument
|
| D | test_manual_embedding_lookup.py | 42 param_shape=(8, 8)): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | apply_came_part4.cc | 70 …auto param_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->GetSh… in CheckInputsShapePart4() local
144 ShapeVector param_shape = out_shape; in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_fused_ada_factor_op.py | 23 param_shape = [2, 3, 2] variable
|
| /third_party/rust/rust/src/tools/rustfmt/src/ |
| D | closures.rs | 280 let param_shape = nested_shape.offset_left(1)?.visual_indent(0); in rewrite_closure_fn_decl() localVariable
304 let param_shape = match tactic { in rewrite_closure_fn_decl() localVariable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | rgb_to_hsv.cc | 86 auto param_shape = param->GetTensorShape(); in CheckParam() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/ |
| D | shard.py | 220 def _check_layout_is_valid(param_name, param_shape, param_strategy): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/ |
| D | gather_v2_ds_fission.cc | 43 auto param_shape = common::AnfAlgo::GetPrevNodeOutputInferShape(origin_node, 0); in CheckInputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | inner_ops.py | 577 … learning_rate_shape, grad_shape, param_shape, exp_avg_shape, exp_avg_sq_row_shape, argument
599 … learning_rate_shape, grad_shape, param_shape, exp_avg_shape, exp_avg_sq_row_shape, argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/auto_parallel/ |
| D | operator_costmodel_test.cc | 169 Shape param_shape = {32}; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/ |
| D | basic_op_infer_shape.cc | 249 auto param_shape = param->Shape(); in HasDynamicShapeInput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/shard/ |
| D | shard.cc | 250 auto param_shape = common::AnfAlgo::GetOutputInferShape(parameter, 0); in SetInputLayout() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/ |
| D | custom_ascend_utils.cc | 165 auto param_shape = GetValue<ShapeVector>(value_ptr_list[i + 2]); in GetZeroValueRefDatas() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/symbol_engine/ |
| D | symbol_engine_impl.cc | 431 auto param_shape = param_abs->GetSymbolicShape(); in GeneralizeParamShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | step_assigned_parallel.cc | 598 auto param_shape = parameter->Shape(); in SetParallelShape() local
|
| D | parameter_manager.cc | 1532 Shape param_shape = inputs_shape[0]; in GenerateTensorLayoutForParamReshapeWithStra() local
|
| D | step_parallel.cc | 1603 auto param_shape = parameter->Shape(); in SetParallelShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | gather_info.cc | 1099 auto param_shape = inputs_shape_[0]; in CheckStrategy() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/trt_pass/ |
| D | trt_op_converter.cc | 931 auto param_shape = Convert2SizeTClipNeg(input_shape); in MS_TRT_CONVERTER_FUNC_REG() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/ |
| D | session_basic.cc | 487 const auto ¶m_shape = common::AnfAlgo::GetOutputInferShape(param, 0); in GetParameterIndex() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/ |
| D | convert.cc | 2010 const auto ¶m_shape = param->Shape(); in SetGraphInputs() local
2299 const auto ¶m_shape = para->Shape(); in GenFakeGraphInRefMode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | operator_costmodel.cc | 1591 auto param_shape = inputs[0].slice_shape(); in GetForwardCommCost() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/ |
| D | rec_generate_strategy.cc | 2216 auto param_shape = ops_[user.first]->inputs_shape()[user.second]; in SetParamStrategy() local
|