Searched refs:split_output (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/test/inductor/ |
| D | test_split_cat_fx_passes.py | 314 split_output = list(torch.split(x, 4, dim=1))
316 [torch.ones(2, 5, 32, 16)] + split_output + [torch.ones(2, 6, 32, 16)],
321 split_output = list(torch.split(x, 4, dim=1))
324 + split_output
330 split_output = list(torch.split(x, 4, dim=2))
332 [torch.ones(2, 32, 5, 16)] + split_output + [torch.ones(2, 32, 6, 16)],
338 split_output = list(torch.split(x, 4, dim=1))
340 [torch.ones(2, 4, 32, 16)] + split_output + [torch.ones(2, 4, 32, 16)],
345 split_output = list(torch.split(x, 4, dim=1))
347 [torch.ones(2, 4, 32, 16)] + split_output + [torch.ones(2, 4, 32, 16)],
[all …]
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Reduce.cuh | 118 int split_output(int parallelism) { in split_output() function
1091 config.output_mult[0] = config.split_output(block_width); in setReduceConfig()
1104 config.output_mult[1] = config.split_output(block_height); in setReduceConfig()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1571 std::string split_output = base + "split"; in ConvertLstmCellOperator() local
1574 split_op->set_name(split_output); in ConvertLstmCellOperator()
1585 *tanh_0_op->add_input() = split_output + ":1"; in ConvertLstmCellOperator()
1592 *logistic_1_op->add_input() = split_output; in ConvertLstmCellOperator()
1607 *logistic_2_op->add_input() = split_output + ":2"; in ConvertLstmCellOperator()
1614 *logistic_3_op->add_input() = split_output + ":3"; in ConvertLstmCellOperator()
|