Home
last modified time | relevance | path

Searched refs:split_input (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/tensorexpr/
H A Delementwise.py21 split_input = True variable in ElementBench
52 if self.split_input:
88 if self.split_input:
98 if self.split_input:
142 for split_input, binary_op in itertools.product([True, False], binary_op_list):
149 split_str = "split" if split_input else "shared"
155 bm_cls.split_input = split_input
158 for split_input, unary_op in itertools.product([True, False], unary_op_list):
165 split_str = "split" if split_input else "shared"
171 bm_cls.split_input = split_input
H A Dbroadcast.py170 split_input = True variable in BroadcastBench
202 if self.split_input:
240 if self.split_input:
247 if self.split_input:
287 for split_input, binary_op in itertools.product([True, False], binary_op_list):
294 split_str = "split" if split_input else "shared"
300 bm_cls.split_input = split_input
303 for split_input, unary_op in itertools.product([True, False], unary_op_list):
310 split_str = "split" if split_input else "shared"
316 bm_cls.split_input = split_input
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dsplit_cat.py167 split_input, split_size, split_dim = _get_split_args(split_node)
168 if split_input is None or split_dim is None or split_size is None:
181 split_dim += split_input.meta["example_value"].dim()
183 new_args = (split_input, split_sections)
227 split_input, split_size, split_dim = _get_split_args_default(split_node)
228 if split_input is None or split_dim is None or split_size is None:
247 next_user.replace_input_with(user, split_input)
829 split_input = split_node.args[0]
832 split_items = [split_input]
838 split_input,
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReduce.cuh112 int split_input(int parallelism) { in split_input() function
1088 config.input_mult[0] = config.split_input(block_width); in setReduceConfig()
1101 config.input_mult[1] = config.split_input(block_height); in setReduceConfig()
1126 config.input_mult[2] = config.split_input(config.ctas_per_output); in setReduceConfig()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_padder.cc319 HloInstruction* zero, bool split_input) { in GenerateBinaryMask() argument
321 split_input ? reshape->operand(0)->shape() : reshape->shape(); in GenerateBinaryMask()
323 split_input ? reshape->shape() : reshape->operand(0)->shape(); in GenerateBinaryMask()