| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | fake_learned_scale_quant_grad_unify_mindir.cc | 50 auto shapes = {AnfAlgo::GetOutputInferShape(lsq_perlayer_grad_node, 0), in CreateOutputsOfLSQPerLayerGradD() local
82 auto shapes = {AnfAlgo::GetOutputInferShape(lsq_perlayer_grad_node, 1)}; in CreateOutputsOfLSQPerLayerReduceGrad() local
107 auto shapes = {AnfAlgo::GetOutputInferShape(lsq_perchannel_grad_node, 0), in CreateOutputsOfLSQPerChannelGradD() local
140 auto shapes = {AnfAlgo::GetOutputInferShape(lsq_perchannel_grad_node, 1)}; in CreateOutputsOfLSQPerChannelReduceGrad() local
|
| D | bn_grad_unify_mindir.cc | 51 …auto shapes = {AnfAlgo::GetOutputInferShape(bn_grad_node, 0), AnfAlgo::GetOutputInferShape(bn_grad… in CreateNewBatchNormGrad() local
|
| D | all_to_all_unify_mindir.cc | 83 std::vector<std::vector<size_t>> shapes(split_count, shape); in CreateSplitNode() local
111 std::vector<std::vector<size_t>> shapes(split_count, single_shape); in CreateAllToAllvNode() local
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_rl_buffer_op.py | 29 def __init__(self, capcity, shapes, types): argument
42 def __init__(self, capcity, shapes, types): argument
55 def __init__(self, capcity, batch_size, shapes, types): argument
|
| D | test_rl_buffer_net.py | 28 def create_tensor(capcity, shapes, dtypes): argument
36 def __init__(self, batch_size, capcity, shapes, types): argument
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_rl_buffer_op.py | 29 def __init__(self, capcity, shapes, types): argument
42 def __init__(self, capcity, shapes, types): argument
55 def __init__(self, capcity, batch_size, shapes, types): argument
|
| D | test_rl_buffer_net.py | 28 def create_tensor(capcity, shapes, types): argument
37 def __init__(self, batch_size, capcity, shapes, types): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | sparse_apply_ftrl_ps_kernel.cc | 26 …const CNodePtr &cnode, const std::shared_ptr<std::vector<std::shared_ptr<std::vector<size_t>>>> &s… in InitKernel()
92 void SparseApplyFtrlPSKernel::ReInit(const std::vector<std::vector<size_t>> &shapes) { in ReInit()
|
| D | sparse_apply_adam_ps_kernel.cc | 29 …const CNodePtr &cnode, const std::shared_ptr<std::vector<std::shared_ptr<std::vector<size_t>>>> &s… in InitKernel()
81 void SparseApplyAdamPSKernel::ReInit(const std::vector<std::vector<size_t>> &shapes) { in ReInit()
|
| D | sparse_apply_lazy_adam_ps_kernel.cc | 29 …const CNodePtr &cnode, const std::shared_ptr<std::vector<std::shared_ptr<std::vector<size_t>>>> &s… in InitKernel()
82 void SparseApplyLazyAdamPSKernel::ReInit(const std::vector<std::vector<size_t>> &shapes) { in ReInit()
|
| D | embedding_look_up_ps_kernel.cc | 32 const std::shared_ptr<std::vector<std::shared_ptr<std::vector<size_t>>>> &shapes) { in InitKernel()
70 void EmbeddingLookUpPSKernel::ReInit(const std::vector<std::vector<size_t>> &shapes) { in ReInit()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | dshape.h | 100 explicit SequeueShape(const BaseShapePtrList &shapes) : p_shapes_(shapes) {} in SequeueShape()
125 explicit TupleShape(const BaseShapePtrList &shapes) : SequeueShape(shapes) {} in TupleShape()
140 explicit ListShape(const BaseShapePtrList &shapes) : SequeueShape(shapes) {} in ListShape()
|
| /third_party/gettext/gettext-tools/src/ |
| D | xg-arglist-callshape.c | 185 struct callshapes *shapes = XMALLOC (struct callshapes); in insert_keyword_callshape() local
222 struct callshapes *shapes = in insert_keyword_callshape() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | batch_norm_grad_split.cc | 46 …auto shapes = {AnfAlgo::GetOutputInferShape(bn_grad_node, 1), AnfAlgo::GetOutputInferShape(bn_grad… in CreateOutputsOfUpdateGrad() local
80 auto shapes = {AnfAlgo::GetOutputInferShape(bn_grad_node, 0)}; in CreateOutputsOfReduceGrad() local
|
| D | layer_norm_grad_split.cc | 50 auto shapes = {AnfAlgo::GetOutputDetailShape(layer_norm_grad, 0), in CreateOutputsOfLayerNormXBackpropV2() local
82 auto shapes = {AnfAlgo::GetOutputDetailShape(layer_norm_grad, kLayerNormGradOutputGammaIndex), in CreateOutputsOfLayerNormBetaGammaBackpropV2() local
|
| D | bn_grad_split.cc | 47 …auto shapes = {AnfAlgo::GetOutputInferShape(bn_grad_node, 1), AnfAlgo::GetOutputInferShape(bn_grad… in CreateOutputsOfUpdateGrad() local
79 auto shapes = {AnfAlgo::GetOutputInferShape(bn_grad_node, 0)}; in CreateOutputsOfReduceGrad() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | square_sum_fusion.cc | 48 auto shapes = {AnfAlgo::GetOutputInferShape(sum, 0)}; in GenerateSquareSumV1() local
69 auto shapes = {AnfAlgo::GetOutputInferShape(sum, 0), AnfAlgo::GetOutputInferShape(square, 0)}; in GenerateSquareSumV2() local
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | config_manager.h | 41 … const std::vector<std::vector<int64_t>> &shapes, const std::vector<int64_t> &input_indexes) in DatasetGraphParam()
61 std::vector<std::vector<int64_t>> shapes() const { return shapes_; } in shapes() function
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info.h | 35 …virtual void ComputeMean(const std::vector<std::vector<size_t>> &shapes, size_t n, size_t server_n… in ComputeMean()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | relu_v2_pass.cc | 86 auto shapes = {AnfAlgo::GetOutputInferShape(relu, 0), mask_shape}; in CreateReluV2() local
108 std::vector<std::vector<size_t>> shapes; in CreateReluGradV2() local
|
| D | apply_momentum_weight_fusion.cc | 63 auto shapes = {AnfAlgo::GetOutputInferShape(node, 0)}; in Process() local
|
| /third_party/mindspore/tests/st/data_transfer/ |
| D | test_tdt_data_transfer.py | 67 def convert_type(shapes, types): argument
95 def __init__(self, shapes): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/data/ |
| D | dataset_utils.cc | 36 void GetShapeAndType(const CNodePtr &kernel_node, std::vector<std::vector<int>> *shapes, std::vecto… in GetShapeAndType()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel_utils.cc | 84 Shapes shapes; in GetValueListShape() local
107 Shapes shapes; in GetNodeShape() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/pyfunc/ |
| D | py_func_cpu_kernel.h | 36 std::vector<std::vector<int64_t>> shapes; member
|