Home
last modified time | relevance | path

Searched defs:is_inplace (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DElementwiseOperationsFixture.h281 …hape &shape1, DataType data_type0, DataType data_type1, DataType output_data_type, bool is_inplace) in setup()
294 …Shape &shape, DataType data_type0, DataType data_type1, DataType output_data_type, bool is_inplace) in setup()
307 …hape &shape1, DataType data_type0, DataType data_type1, DataType output_data_type, bool is_inplace) in setup()
320 …Shape &shape, DataType data_type0, DataType data_type1, DataType output_data_type, bool is_inplace) in setup()
333 …pe0, DataType data_type1, DataType output_data_type, ActivationLayerInfo act_info, bool is_inplace) in setup()
346 …pe0, DataType data_type1, DataType output_data_type, ActivationLayerInfo act_info, bool is_inplace) in setup()
359 …pe0, DataType data_type1, DataType output_data_type, ActivationLayerInfo act_info, bool is_inplace) in setup()
373 … QuantizationInfo qinfo0, QuantizationInfo qinfo1, QuantizationInfo qinfo_out, bool is_inplace) in setup()
387 …hape &shape1, DataType data_type0, DataType data_type1, DataType output_data_type, bool is_inplace) in setup()
400 …Shape &shape, DataType data_type0, DataType data_type1, DataType output_data_type, bool is_inplace) in setup()
[all …]
H A DArithmeticOperationsFixture.h50 …QuantizationInfo qinfo1, QuantizationInfo qinfo_out, ActivationLayerInfo act_info, bool is_inplace) in setup()
154 …ape0, const TensorShape &shape1, DataType data_type, ConvertPolicy convert_policy, bool is_inplace) in setup()
166 … setup(const TensorShape &shape, DataType data_type, ConvertPolicy convert_policy, bool is_inplace) in setup()
178 …1, DataType data_type, ConvertPolicy convert_policy, ActivationLayerInfo act_info, bool is_inplace) in setup()
190 …e, DataType data_type, ConvertPolicy convert_policy, ActivationLayerInfo act_info, bool is_inplace) in setup()
202 …icy, QuantizationInfo qinfo0, QuantizationInfo qinfo1, QuantizationInfo qinfo_out, bool is_inplace) in setup()
216 bool is_inplace) in setup()
228 …ape0, const TensorShape &shape1, DataType data_type, ConvertPolicy convert_policy, bool is_inplace) in setup()
241 bool is_inplace) in setup()
253 … setup(const TensorShape &shape, DataType data_type, ConvertPolicy convert_policy, bool is_inplace) in setup()
[all …]
H A DPixelWiseMultiplicationFixture.h60 bool is_inplace) in setup()
159 …dt_out, float scale, ConvertPolicy convert_policy, RoundingPolicy rounding_policy, bool is_inplace) in setup()
172 bool is_inplace) in setup()
184 …licy convert_policy, RoundingPolicy rounding_policy, ActivationLayerInfo act_info, bool is_inplace) in setup()
196 …licy convert_policy, RoundingPolicy rounding_policy, ActivationLayerInfo act_info, bool is_inplace) in setup()
209 ActivationLayerInfo act_info, bool is_inplace) in setup()
222 … QuantizationInfo qinfo0, QuantizationInfo qinfo1, QuantizationInfo qinfo_out, bool is_inplace) in setup()
235 … QuantizationInfo qinfo0, QuantizationInfo qinfo1, QuantizationInfo qinfo_out, bool is_inplace) in setup()
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/operators/
H A DMulFixture.h204 void setup(const TensorShape &shape0, DataType data_type, bool is_inplace) in setup()
215 …id setup(const TensorShape &shape0, const TensorShape &shape1, DataType data_type, bool is_inplace) in setup()
226 … TensorShape &shape1, const TensorShape &shape2, DataType data_type, bool is_inplace, bool fuse_tw… in setup()
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/
H A DElementwiseBinaryFixture.h224 …d setup(ArithmeticOperation ref_op, const TensorShape &shape0, DataType data_type, bool is_inplace) in setup()
235 … ref_op, const TensorShape &shape0, const TensorShape &shape1, DataType data_type, bool is_inplace) in setup()
246 … TensorShape &shape1, const TensorShape &shape2, DataType data_type, bool is_inplace, bool fuse_tw… in setup()
/aosp_15_r20/external/pytorch/test/
H A Dtest_foreach.py247 is_inplace, argument
489 is_inplace, argument
1489 def check_autodiff_sample(op, sample, dtype, is_inplace): argument
H A Dtest_ops_fwd_gradients.py54 def _forward_grad_helper(self, device, dtype, op, variant, is_inplace): argument
H A Dtest_namedtensor.py709 def test_mixed_unnamed_named(op, is_inplace): argument
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcomposite_compliance.py97 def is_inplace(func): function
H A Dcommon_utils.py5104 def is_inplace(variant): function
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchedFallback.cpp75 static void warnFallback(const c10::FunctionSchema& schema, bool is_inplace, bool is_nested=false) { in warnFallback()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_function.cpp1010 bool is_inplace, in _trace_post_record()
1116 bool is_inplace = static_cast<bool>(grad_fn->dirty_tensors); in process_outputs() local
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_ops.py344 def is_inplace(op, variant): function
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py209 def __init__(self, binary_op, ibinary_op, is_inplace, is_scalar): argument
228 def __init__(self, binary_op, ibinary_op, is_inplace, is_scalar): argument
241 def __init__(self, binary_op, ibinary_op, is_inplace, relu_callable, argument