Home
last modified time | relevance | path

Searched refs:num_ops (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/tests/ut/python/dataset/
Dtest_uniform_augment.py31 def test_uniform_augment_callable(num_ops=2): argument
45 uni_aug = C.UniformAugment(transforms=transforms_ua, num_ops=num_ops)
50 def test_uniform_augment(plot=False, num_ops=2): argument
89 num_ops=num_ops),
114 def test_cpp_uniform_augment(plot=False, num_ops=2): argument
146 uni_aug = C.UniformAugment(transforms=transforms_ua, num_ops=num_ops)
173 def test_cpp_uniform_augment_exception_pyops(num_ops=2): argument
187 C.UniformAugment(transforms=transforms_ua, num_ops=num_ops)
193 def test_cpp_uniform_augment_exception_large_numops(num_ops=6): argument
206 _ = C.UniformAugment(transforms=transforms_ua, num_ops=num_ops)
[all …]
Dtest_profiling.py73 def confirm_ops_in_pipeline(num_ops, op_list): argument
81 assert len(op_info) == num_ops
82 for i in range(num_ops):
Dtest_serdes_dataset.py428 vision.UniformAugment(transforms=transforms_ua, num_ops=5)]
/third_party/grpc/test/core/end2end/tests/
Dcancel_with_status.cc38 size_t num_ops, in begin_test() argument
43 config.name, num_ops); in begin_test()
93 grpc_end2end_test_fixture f, size_t num_ops) { in simple_request_body() argument
104 gpr_log(GPR_DEBUG, "test with %" PRIuPTR " ops", num_ops); in simple_request_body()
138 GPR_ASSERT(num_ops <= (size_t)(op - ops)); in simple_request_body()
139 error = grpc_call_start_batch(c, ops, num_ops, tag(1), nullptr); in simple_request_body()
165 size_t num_ops) { in test_invoke_simple_request() argument
168 f = begin_test(config, "test_invoke_simple_request", num_ops, nullptr, in test_invoke_simple_request()
170 simple_request_body(config, f, num_ops); in test_invoke_simple_request()
Dnegative_deadline.cc90 grpc_end2end_test_fixture f, size_t num_ops) { in simple_request_body() argument
101 gpr_log(GPR_DEBUG, "test with %" PRIuPTR " ops", num_ops); in simple_request_body()
135 GPR_ASSERT(num_ops <= (size_t)(op - ops)); in simple_request_body()
136 error = grpc_call_start_batch(c, ops, num_ops, tag(1), nullptr); in simple_request_body()
154 size_t num_ops) { in test_invoke_simple_request() argument
158 simple_request_body(config, f, num_ops); in test_invoke_simple_request()
Dcancel_before_invoke.cc35 size_t num_ops, in begin_test() argument
40 config.name, num_ops); in begin_test()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/ir/vision/
Duniform_aug_ir.cc33 int32_t num_ops) in UniformAugOperation() argument
34 : transforms_(transforms), num_ops_(num_ops) {} in UniformAugOperation()
85 int32_t num_ops = op_params["num_ops"]; in from_json() local
86 *operation = std::make_shared<vision::UniformAugOperation>(transforms, num_ops); in from_json()
Duniform_aug_ir.h40 …formAugOperation(const std::vector<std::shared_ptr<TensorOperation>> &transforms, int32_t num_ops);
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/
Duniform_aug_op.cc23 UniformAugOp::UniformAugOp(std::vector<std::shared_ptr<TensorOp>> op_list, int32_t num_ops) in UniformAugOp() argument
24 : tensor_op_list_(std::move(op_list)), num_ops_(num_ops) { in UniformAugOp()
Duniform_aug_op.h38 UniformAugOp(std::vector<std::shared_ptr<TensorOp>> op_list, int32_t num_ops);
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dgraph_info.cc74 void SetOpsNumToExecutor(size_t num_ops) { in SetOpsNumToExecutor() argument
76 executor->SetNumOpsInfo(num_ops); in SetOpsNumToExecutor()
/third_party/mindspore/mindspore/dataset/vision/
Dvalidators.py753 [transforms, num_ops], _ = parse_user_args(method, *args, **kwargs)
754 type_check(num_ops, (int,), "num_ops")
755 check_positive(num_ops, "num_ops")
757 if num_ops > len(transforms):
833 [transforms, num_ops], _ = parse_user_args(method, *args, **kwargs)
843 type_check(num_ops, (int,), "num_ops")
844 check_positive(num_ops, "num_ops")
845 if num_ops > len(transforms):
Dpy_transforms.py1691 def __init__(self, transforms, num_ops=2): argument
1693 self.num_ops = num_ops
1706 return util.uniform_augment(img, self.transforms.copy(), self.num_ops)
Dc_transforms.py1821 def __init__(self, transforms, num_ops=2): argument
1823 self.num_ops = num_ops
1832 return cde.UniformAugOperation(transforms, self.num_ops)
Dpy_transforms_util.py1554 def uniform_augment(img, transforms, num_ops): argument
1571 op_idx = np.random.choice(len(transforms), size=num_ops, replace=False)
/third_party/mindspore/tests/ut/python/parallel/
Dtest_auto_parallel_for_loop.py126 num_ops = _cell_graph_executor._get_num_parallel_ops(net)
128 assert num_ops == expected_num
Dtest_auto_parallel_for_loop_reshape.py133 num_ops = _cell_graph_executor._get_num_parallel_ops(net)
135 assert num_ops == expected_num
Dtest_auto_parallel_for_loop_multi_subgraph.py133 num_ops = _cell_graph_executor._get_num_parallel_ops(net)
135 assert expected_num == num_ops
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/
Dvision.h1018 explicit UniformAugment(const std::vector<TensorTransform *> &transforms, int32_t num_ops = 2);
1023 …iformAugment(const std::vector<std::shared_ptr<TensorTransform>> &transforms, int32_t num_ops = 2);
1028 …gment(const std::vector<std::reference_wrapper<TensorTransform>> &transforms, int32_t num_ops = 2);
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Dvision.cc1008 UniformAugment::UniformAugment(const std::vector<TensorTransform *> &transforms, int32_t num_ops) in UniformAugment() argument
1013 data_->num_ops_ = num_ops; in UniformAugment()
1016 …t::UniformAugment(const std::vector<std::shared_ptr<TensorTransform>> &transforms, int32_t num_ops) in UniformAugment() argument
1022 data_->num_ops_ = num_ops; in UniformAugment()
1025 …ormAugment(const std::vector<std::reference_wrapper<TensorTransform>> &transforms, int32_t num_ops) in UniformAugment() argument
1029 data_->num_ops_ = num_ops; in UniformAugment()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_auto_parallel.cc957 auto num_ops = entire_costgraph->GetOperators().size(); in ParallelStrategySearch() local
958 SetOpsNumToExecutor(num_ops); in ParallelStrategySearch()
960 …MS_LOG(INFO) << "After the augmenting procedure, there are " << num_ops << " operators, and " << n… in ParallelStrategySearch()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/kernels/ir/image/
Dbindings.cc625 .def(py::init([](const py::list transforms, int32_t num_ops) { in __anon574f76b45c02() argument
627 … std::make_shared<vision::UniformAugOperation>(std::move(toTensorOperations(transforms)), num_ops); in __anon574f76b45c02()
/third_party/skia/third_party/externals/swiftshader/third_party/llvm-10.0/llvm/lib/Target/AMDGPU/
DAMDGPUPrintfRuntimeBinding.cpp58 StringRef fmt, size_t num_ops) const;
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline.cc462 void GraphExecutorPy::SetNumOpsInfo(size_t num_ops) { in SetNumOpsInfo() argument
464 phase_to_num_op_info_[phase_] = num_ops; in SetNumOpsInfo()