Searched defs:pad_shape (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/data/ |
| D | pad_end_op.h | 30 explicit PadEndOp(const TensorShape &pad_shape, const std::shared_ptr<Tensor> &pad_value) in PadEndOp()
|
| D | data_utils.cc | 376 … std::shared_ptr<Tensor> &src, std::shared_ptr<Tensor> *dst, const std::vector<dsize_t> &pad_shape, in PadEnd()
402 const std::vector<dsize_t> &pad_shape, float pad_val) { in PadEndNumeric()
461 const std::vector<dsize_t> &pad_shape, const std::string &pad_val) { in PadEndString()
|
| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_pad_end_op.py | 27 def pad_compare(array, pad_shape, pad_value, res): argument
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | pad.cc | 67 auto pad_shape = in_tensors_.at(1)->shape(); in CheckSpecs() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hccl_kernel_metadata.cc | 51 auto pad_shape = trans::PaddingShapeTo4dDefault(input_shape); in GetKernelFormat() local
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | bucket_batch_by_length_op.cc | 136 std::vector<dsize_t> pad_shape = pair.second.first.AsVector(); in PadAndBatchBucket() local
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/kernels/ir/ |
| D | bindings.cc | 139 .def(py::init([](TensorShape pad_shape, const std::shared_ptr<Tensor> &pad_value) { in __anon3fa898e21002()
|
| /third_party/mindspore/mindspore/dataset/transforms/ |
| D | c_transforms.py | 310 def __init__(self, pad_shape, pad_value=None): argument
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/ir/data/ |
| D | transforms_ir.cc | 192 PadEndOperation::PadEndOperation(const TensorShape &pad_shape, const std::shared_ptr<Tensor> &pad_v… in PadEndOperation()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/ |
| D | transforms.cc | 169 PadEnd::PadEnd(const std::vector<dsize_t> &pad_shape, const MSTensor &pad_value) in PadEnd()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.cc | 293 …ShapeVector pad_shape = trans::PaddingShape(out_shape, AnfAlgo::GetOutputFormat(input, 0), padding… in NewTransOpNode() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | insert_pad.cc | 150 const vec &pad_shape, const vec &tail_shape) { in InsertPad()
|