| /third_party/mindspore/mindspore/ops/operations/ |
| D | rl_ops.py | 126 def infer_shape(self, data_shape, count_shape, head_shape): argument
207 def infer_shape(self, data_shape, exp_shape, count_shape, head_shape): argument
309 def infer_shape(self, data_shape, count_shape, head_shape, index_shape): argument
|
| D | _thor_ops.py | 330 def infer_shape(self, data_shape): argument
|
| D | _inner_ops.py | 1074 def check_shape(self, indices_shape, data_shape): argument
|
| D | _quant_ops.py | 1764 def infer_shape(self, data_shape, data_min_shape, data_max_shape, cumsum_shape): argument
|
| /third_party/mindspore/mindspore/ops/composite/multitype_ops/ |
| D | _constexpr_utils.py | 140 def make_tensor(a, dtype=mstype.int64, data_shape=None, dim_size=-1): argument
194 def get_source_shape(data_shape, value_shape): argument
473 def convert_scalar_to_tensor(data_shape, data_dtype, indices_shape, value, op_type): argument
483 def generate_updates_shape(data_shape, index_shape, op_type): argument
564 def get_stride_info_from_slice(data_shape, slice_index): argument
578 def get_stride_info_from_integer(data_shape, number): argument
604 def get_stride_info_from_tuple(data_shape, tuple_index): argument
|
| D | _compile_utils.py | 873 def remove_expanded_dims(tuple_index, data_shape, value): argument
933 def format_index(idx, data_shape, cur_dim): argument
|
| /third_party/mindspore/tests/st/ops/custom_ops_tbe/ |
| D | cus_square.py | 33 def infer_shape(self, data_shape): argument
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | group_depthwise_op_convert_pass.cc | 67 auto data_shape = data_shape_ptr->shape(); in Run() local
|
| D | node_infershape.cc | 570 std::vector<int64_t> data_shape; in ConvertTensorListToAbstract() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_kernel_runtime.cc | 111 ShapeVector data_shape = tensor->shape(); in AssignValueNodeAddress() local
228 ShapeVector data_shape = tensor->shape(); in CreatTensorForOutput() local
335 ShapeVector data_shape = tensor->shape(); in BindInputTensorAddressPtr() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_conv_parser.cc | 141 auto data_shape = data_tensor->shape; in Parse() local
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ascend/ |
| D | ascend_ps_cache.cc | 41 bool SetProtoInputs(const std::vector<std::vector<size_t>> &data_shape, const std::vector<TypeId> &… in SetProtoInputs()
66 bool SetProtoOutputs(const std::vector<std::vector<size_t>> &data_shape, const std::vector<TypeId> … in SetProtoOutputs()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tf_lstm_cell_fusion.cc | 247 auto data_shape = origin_tensor->shape(); in SplitWeights() local
302 auto data_shape = origin_tensor->shape(); in PopulateBiasNode() local
|
| D | tflite_lstm_cell_fusion.cc | 446 auto data_shape = origin_tensor->shape(); in GetConcatedParam() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_attrs_process.cc | 118 std::vector<size_t> data_shape = AnfAlgo::GetInputDeviceShape(bn_grad1, 0); in SetAkgAttrsForBNGrad2() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | slice_base.c | 22 int32_t data_shape[DIMENSION_8D]; in PadSliceParameterTo8D() local
|