Searched defs:params_shape (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | gather.cc | 68 bool GetBatchDimByInput(std::vector<int64_t> *params_shape, std::vector<int64_t> *indices_shape, in… in GetBatchDimByInput()
83 auto params_shape = input_args[kInputIndex0]->GetShape()->GetShapeVector(); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/pynative/ |
| D | test_pynative_embeddinglookup.py | 56 …def __init__(self, params_shape, indices_shape, offset=0, low=0, high=2, dtype=np.float32, ids_typ… argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | embedding_lookup.cc | 85 auto params_shape = params_shape_ptr->shape(); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/operator/ |
| D | custom_op_infer.cc | 74 auto params_shape = params_shape_ptr->shape(); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | norm_fusion.cc | 152 … const std::vector<int> &mean_axes, const std::vector<int> ¶ms_shape, in GetNormTypeAndAxis()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | gather_info.cc | 169 auto params_shape = inputs_shape_.at(0); in GetAttrs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | cache_embedding.cc | 388 auto params_shape = params_shp->shape(); in CreateUpdateCache() local
|