| /external/tensorflow/tensorflow/core/kernels/ |
| D | typed_conditional_accumulator_base.h | 39 const string& reduction_type) in TypedConditionalAccumulatorBase()
|
| D | conditional_accumulator.h | 56 const string& name, const string& reduction_type) in ConditionalAccumulator()
|
| D | conditional_accumulator_base.cc | 23 const string& reduction_type) in ConditionalAccumulatorBase()
|
| D | sparse_conditional_accumulator.h | 53 const string& name, const string& reduction_type) in SparseConditionalAccumulator()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/python/ |
| D | tflite_text_api.py | 57 reduction_type=None, argument
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | triton_split_scan.py | 78 def reduction(self, dtype, src_dtype, reduction_type, value): argument
|
| D | cpp.py | 158 def reduction_init(reduction_type, dtype): argument
190 def reduction_acc_type(reduction_type, dtype): argument
200 reduction_type, argument
234 def reduction_project(reduction_type, acc): argument
1742 reduction_type, argument
1920 def reduction(self, dtype, src_dtype, reduction_type, value): argument
2539 def reduction(self, dtype, src_dtype, reduction_type, value): argument
2772 def reduction_init_vec(self, reduction_type, dtype): argument
2806 def reduction_acc_type_vec(self, reduction_type, dtype): argument
2836 reduction_type, argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reduction_ops_common.cc | 31 DataType reduction_type) in XlaReductionOp()
|
| D | pooling_ops.cc | 49 const DataType reduction_type) in PoolingOp()
493 auto reduction_type = XlaHelpers::SumAccumulationType(ctx->input_type(1)); in Compile() local
|
| /external/pytorch/torch/_inductor/codegen/cuda/ |
| D | cutlass_epilogue_gen.py | 200 def reduction(self, dtype, src_dtype, reduction_type, value): argument
357 def reduction(self, dtype, src_dtype, reduction_type, value): argument
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| D | ngrams.cc | 59 std::string reduction_type; member
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | tftext_utils.cc | 237 auto reduction_type = in CreateNgramsCustomOption() local
|
| /external/pytorch/torch/_inductor/ |
| D | utils.py | 1548 def is_welford_reduction(reduction_type): argument
1552 def reduction_num_outputs(reduction_type): argument
1816 reduction_type, argument
|
| D | loop_body.py | 452 def reduction(self, dtype, src_dtype, reduction_type, value): argument
|
| D | ir.py | 554 def get_reduction_combine_fn(reduction_type, dtype, arg_break_ties_left=True): argument
679 reduction_type, argument
898 def _unroll_reduction_fn(inner_fn, reduction_ranges, reduction_type, src_dtype): argument
1081 def default_accumulator(reduction_type, dtype): argument
1107 def default_value(reduction_type, dtype): argument
1333 def num_reduction_outputs(reduction_type): argument
1347 reduction_type, argument
1499 def default_value(reduction_type, dtype): argument
8077 def reduction(self, dtype, src_dtype, reduction_type, value): argument
|
| D | lowering.py | 3413 def scatter_reduce(x, dim: int, index, src, reduction_type, **kwargs): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | data_flow_ops.py | 1336 reduction_type="MEAN"): argument
1438 reduction_type="MEAN"): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_all_reduce_code_motion.cc | 171 std::optional<ReductionKind> reduction_type = in IsAllReduceMovable() local
|
| /external/pytorch/torch/utils/_sympy/ |
| D | value_ranges.py | 968 def reduction(self, name, dtype, src_dtype, reduction_type, index, value): argument
|
| /external/pytorch/test/ |
| D | test_mps.py | 5208 def helper(n, c, h, w, reduction_type, dtype=torch.float32): argument
|