Home
last modified time | relevance | path

Searched defs:current_shape (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Drandom_data_op.cc115 std::vector<dsize_t> current_shape = current_col.Shape().AsVector(); in CreateRandomRow() local
Dtf_reader_op.cc691 TensorShape current_shape = TensorShape::CreateUnknownRankShape(); in LoadFeature() local
778 TensorShape current_shape = TensorShape::CreateScalar(); in LoadBytesList() local
855 TensorShape current_shape = TensorShape::CreateUnknownRankShape(); in LoadIntList() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dsparse_concat.cc295 const typename TTypes<int64_t>::Vec current_shape = shapes_t[i]; in DoCompute() local
322 std::vector<int64_t> current_shape; in DoCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/parallel_compile/akg_compiler/
Dbuild_tbe_kernel.py163 def _shape_error(current_shape, current_format): argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/
Dparameter.py816 …def _set_data_check_input_valid(current_shape, data_shape, current_tensor_is_init, incoming_tensor… argument