Searched defs:multiples (Results 1 – 13 of 13) sorted by relevance
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_tile.py | 24 def __init__(self, multiples): argument
33 def get_output(x, multiples, enable_graph_kernel=False): argument
40 def test_tile(shape, dtype, multiples): argument
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/ |
| D | caffe_tile_parser.cc | 39 std::vector<int32_t> multiples; in Parse() local
|
| /third_party/mindspore/mindspore/lite/src/ops/compat/v0/ |
| D | tile_compat_v0.cc | 46 std::vector<int> multiples = std::vector<int>(multiples_attr->begin(), multiples_attr->end()); in TransferTileToAttr() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | tile_info.cc | 43 ValueTuplePtr multiples = input_value_[1]->cast<ValueTuplePtr>(); in GetAttrs() local
78 Shapes multiples = {tmp}; in CheckStrategy() local
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/op/ |
| D | tile_npu.cc | 49 std::vector<int> multiples; in SetNPUInputs() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | tile_gpu_kernel.h | 97 std::vector<int64_t> multiples = GetAttr<std::vector<int64_t>>(kernel_node, "multiples"); in Init() local
|
| /third_party/mindspore/tests/vm_impl/ |
| D | array_ops_vm_impl.py | 186 def vm_impl(x, multiples): argument
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 225 def _tile_shape(multiples, shapex): argument
256 def bprop(x, multiples, out, dout): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | sparse_softmax_cross_entropy_with_logits_unify_mindir.cc | 266 auto multiples = MakeValue(multiple_value); in CreateTile() local
|
| /third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 1498 auto multiples = node->input(kTwoNum); in ExportPrimTile() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 1964 def __infer__(self, x, multiples): argument
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | ops_v0_generated.h | 7061 const flatbuffers::Vector<int32_t> *multiples() const { in multiples() function
|
| /third_party/mindspore/mindspore/lite/schema/inner/ |
| D | ops_v0_generated.h | 9995 const flatbuffers::Vector<int32_t> *multiples() const { in multiples() function
|