Home
last modified time | relevance | path

Searched refs:op_list (Results 1 – 25 of 25) sorted by relevance

/third_party/mindspore/mindspore/nn/wrap/
Dgrad_reducer.py43 op_list = ()
48 op_list = op_list + (op,)
49 return op_list
54 op_list = ()
72 op_list = op_list + (op,)
75 op_list = _init_allreduce_operators(len(parameters), split_indices, group)
78 op_list = ()
79 return op_list, param_fusion
382 self.op_list = _init_allreduce_operators(len(parameters), split_indices, group)
385 …self.op_list, param_fusion = _init_allreduce_operators_by_parameters(parameters, split_indices, gr…
[all …]
/third_party/mindspore/tests/ut/python/dataset/
Dtest_compose.py37 def test_config(arr, op_list): argument
40 data = data.map(input_columns=["col"], operations=op_list)
112 def test_config(arr, input_columns, output_cols, op_list): argument
114 data = data.map(operations=op_list, input_columns=input_columns, output_columns=output_cols,
139 def test_config(arr, input_columns, output_cols, op_list): argument
141 data = data.map(operations=op_list, input_columns=input_columns, output_columns=output_cols,
175 def test_config(plot, file_name, op_list): argument
178 data1 = data1.map(operations=op_list, input_columns=["image"])
196 test_config(op_list=[c_vision.Decode(),
202 test_config(op_list=[c_vision.Decode(),
[all …]
Dtest_c_random_choice.py27 def test_config(arr, op_list): argument
30 data = data.map(operations=ops.RandomChoice(op_list), input_columns=["col"])
Dtest_c_compose.py25 def test_config(arr, op_list): argument
28 data = data.map(operations=ops.Compose(op_list), input_columns=["col"])
Dtest_c_random_apply.py24 def test_config(arr, op_list, prob=0.5): argument
27 data = data.map(operations=ops.RandomApply(op_list, prob), input_columns=["col"])
Dtest_profiling.py73 def confirm_ops_in_pipeline(num_ops, op_list): argument
83 assert op_info[i]["op_type"] in op_list
/third_party/mindspore/tests/ut/python/pipeline/parse/
Dtest_for_stmt.py106 self.op_list = ()
110 self.op_list = self.op_list + (op,)
117 fushion_res = self.hyper_map(F.partial(_grad_fusion), ret, self.op_list)
/third_party/mindspore/tests/ut/cpp/dataset/
Dir_callback_test.cc149 std::vector<std::shared_ptr<TensorOp>> op_list; in TEST_F() local
151 op_list.push_back(my_no_op); in TEST_F()
153 …std::make_shared<MapOp>(input_columns, output_columns, std::move(op_list), num_workers, op_connect… in TEST_F()
207 std::vector<std::shared_ptr<TensorOp>> op_list; in TEST_F() local
209 op_list.push_back(my_no_op); in TEST_F()
211 …std::make_shared<MapOp>(input_columns, output_columns, std::move(op_list), num_workers, op_connect… in TEST_F()
280 std::vector<std::shared_ptr<TensorOp>> op_list; in TEST_F() local
282 op_list.push_back(my_no_op); in TEST_F()
284 …std::make_shared<MapOp>(input_columns, output_columns, std::move(op_list), num_workers, op_connect… in TEST_F()
Doptimization_pass_test.cc99 std::vector<std::shared_ptr<TensorOperation>> op_list = {decode, resize}; in TEST_F() local
102 std::shared_ptr<MapNode> map_node = std::make_shared<MapNode>(root, op_list, op_name); in TEST_F()
Dexecute_test.cc378 std::vector<std::shared_ptr<TensorTransform>> op_list = {decode, resize, normalize}; in TEST_F() local
379 mindspore::dataset::Execute Transform(op_list); in TEST_F()
416 std::vector<std::shared_ptr<TensorTransform>> op_list = {decode, resize, centercrop, hwc2chw}; in TEST_F() local
417 mindspore::dataset::Execute Transform(op_list, MapTargetDevice::kCpu); in TEST_F()
/third_party/pixman/test/
Dcombiner-test.c7 static const pixman_op_t op_list[] = variable
128 for (i = 0; i < ARRAY_LENGTH (op_list); ++i) in main()
130 pixman_op_t op = op_list[i]; in main()
Dsolid-test.c35 static const pixman_op_t op_list[] = { variable
284 op = op_list[prng_rand_n (ARRAY_LENGTH (op_list))]; in test_solid()
Dblitters-test.c108 static pixman_op_t op_list[] = { variable
266 op = op_list[prng_rand_n (ARRAY_LENGTH (op_list))]; in test_composite()
Dcover-test.c77 static const pixman_op_t op_list[] = { variable
307 op = op_list[prng_rand_n (ARRAY_LENGTH (op_list))]; in test_cover()
Dstress-test.c823 static const pixman_op_t op_list[] = variable
928 op = op_list [prng_rand_n (ARRAY_LENGTH (op_list))]; in run_test()
963 op = op_list [prng_rand_n (ARRAY_LENGTH (op_list))]; in run_test()
Dutils.c1100 static const operator_entry_t op_list[] = variable
1393 for (i = 0; i < ARRAY_LENGTH (op_list); ++i) in list_operators()
1395 const operator_entry_t *ent = &op_list[i]; in list_operators()
1440 for (i = 0; i < ARRAY_LENGTH (op_list); ++i) in operator_from_string()
1442 const operator_entry_t *ent = &op_list[i]; in operator_from_string()
1480 for (i = 0; i < ARRAY_LENGTH (op_list); ++i) in operator_name()
1482 const operator_entry_t *ent = &op_list[i]; in operator_name()
/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/tests/ut/cpp/parallel/
Dstep_parallel_test.cc403 OperatorVector op_list = {op, op}; in TEST_F() local
418 ForwardCommunication(op_list, cnode); in TEST_F()
442 OperatorVector op_list; in TEST_F() local
458 EXPECT_THROW({ ForwardCommunication(op_list, cnode); }, std::runtime_error); in TEST_F()
465 OperatorVector op_list; in TEST_F() local
484 OperatorVector op_list = {op}; in TEST_F() local
485 EXPECT_THROW({ ForwardCommunication(op_list, cnode); }, std::runtime_error); in TEST_F()
/third_party/mindspore/mindspore/_extends/parallel_compile/tbe_compiler/
Dtbe_helper.py166 op_list = job_content["op_list"]
168 for op in op_list:
/third_party/mindspore/mindspore/nn/layer/
Dbasic.py1435 self.op_list = []
1438 self.op_list.append((inner.Roll(shift=self.shift, axis=0), self.axis))
1445 self.op_list.append((inner.Roll(shift=self.shift[idx], axis=0), self.axis[idx]))
1449 for single_op_roll, single_axis in self.op_list:
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/
Dsingle_tbe_json_creator.cc37 std::vector<nlohmann::json> op_list; in GenJson() local
38 if (!GenOpListJson(anf_node, &op_list)) { in GenJson()
43 (*kernel_json)[kJOpList] = op_list; in GenJson()
/third_party/mindspore/mindspore/profiler/parser/
Dintegrator.py927 op_list = line.strip('\n').strip().split(';')
928 time_arr = op_list[-1]
932 line_list = op_list[:2] + time
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_build.cc982 auto op_list = kernel_json["op_list"]; in GetIOSize2() local
983 for (size_t i = 0; i < op_list.size(); i++) { in GetIOSize2()
984 auto op_info = op_list[i]; in GetIOSize2()
/third_party/quickjs/
Dqjscalc.js58 function operators_set(proto, ...op_list) argument
63 for(i = 0; i < op_list.length; i++) {
64 a = op_list[i];