Home
last modified time | relevance | path

Searched full:add_3 (Results 1 – 25 of 28) sorted by relevance

12

/aosp_15_r20/external/pytorch/test/
H A Dtest_fx_passes.py36 add_3 = add_1 + linear_1
37 add_4 = add_2 + add_3
70 add_3 = add_1 + add_2
71 add_4 = add_1 + relu_1 + add_3
82 add_3 = add_1 + relu_1
83 add_4 = add_1 + add_3
128 add_3 = add_2 + 1
129 return add_3, add_1
140 add_3 = add_2 + add_1
142 return add_3
[all …]
H A Dtest_fx_experimental.py158 add_3 = add_2 + torch.rand(4)
159 return add_3
222 add_3 = add_2 + linear_1
223 return add_3
314 add_3 = add_2 + linear_1
315 add_4 = add_2 + add_3
371 add_3 = add_2 + torch.rand(4)
373 add_5 = add_3 + add_4
492 add_3 = add_2 + linear_1
493 add_4 = add_2 + add_3
[all …]
/aosp_15_r20/external/pytorch/test/edge/
H A Dtest_operator_registration.cpp31 // custom::add_3.out(Tensor a, Tensor b, Tensor c, *, Tensor(a!) out) -> Tensor(a!)
38 ASSERT_TRUE(hasKernelFn("custom::add_3.out")); in TEST()
39 auto op = getKernelFn("custom::add_3.out"); in TEST()
H A Dselected_operators.yaml7 custom::add_3.out:
458 custom::add_3.out:
H A Dcustom_ops.yaml1 - func: custom::add_3.out(Tensor a, Tensor b, Tensor c, *, Tensor(a!) out) -> Tensor(a!)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_rematerialization_test.cc333 // F32[1024] %add_3 = add(%bcast, call_2) in TEST_F()
334 // F32[1024] %call_3 = call(Subcomputation, {%add_3}) in TEST_F()
374 auto add_3 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F() local
377 HloInstruction::CreateCall(vec1024_shape_, {add_3}, subcomputation)); in TEST_F()
399 EXPECT_EQ(add_3->operand(0), bcast); in TEST_F()
414 // The operands of add_2, add_3, and add_4 should all be rematerialized in TEST_F()
418 EXPECT_NE(add_3->operand(0), bcast); in TEST_F()
419 EXPECT_THAT(add_3->operand(0), op::Broadcast(param)); in TEST_F()
H A Dhlo_ordering_test.cc391 HloInstruction* add_3 = FindInstruction(module.get(), "add.3"); in TEST_F() local
402 dataflow->GetValueDefinedAt(add_3))); in TEST_F()
404 dataflow->GetValueDefinedAt(add_3))); in TEST_F()
H A Dreduce_decomposer_test.cc59 …] = f32[5,4]{1,0} reduce([[p_1:%[^ ]+]], [[z_2:%[^ ]+]]), dimensions={1}, to_apply=[[add_3:%[^ ]+]] in TEST_F()
/aosp_15_r20/external/pytorch/test/onnx/expect/
H A DTestOperators.test_gelu.expect27 output: "onnx::Add_3"
45 input: "onnx::Add_3"
H A DTestOperators.test_chunk.expect28 output: "onnx::Add_3"
66 input: "onnx::Add_3"
H A DTestOperators.test_aten_embedding_2.expect8 output: "onnx::Add_3"
29 input: "onnx::Add_3"
H A DTestOperators.test_scatter_add_opset11.expect37 name: "Add_3"
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_experimental.py65 add_3 = torch.ops.aten.add.Tensor(sum_1, sum_2); sum_1 = sum_2 = None
67 add_4 = torch.ops.aten.add.Tensor(add_3, sum_3); add_3 = sum_3 = None
H A Dtest_torchbind.py716 add_3 = torch.ops.aten.add.Tensor(add_2, x); add_2 = x = None
717 return (getitem_8, add_3, add_1, tq)""", # noqa: B950
1323 add_3 = torch.ops.aten.add.Tensor(getitem_1, x); getitem_1 = x = None
1324 return (getitem, add_3)""", # noqa: B950
H A Dtest_passes.py927 add_3 = torch.ops.aten.add.Tensor(cos, 1); cos = None
928 return (add_2, add_3)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dtree_reduction_rewriter_test.cc223 …,100]{1,0} reduce([[input_1:%[^ ]+]], [[zero_2:%[^ ]+]]), dimensions={2}, to_apply=[[add_3:%[^ ]+]] in TEST_F()
224 …out_1_4:%[^ ]+]] = f32[100]{0} reduce([[reduce_0]], [[zero_2]]), dimensions={0}, to_apply=[[add_3]] in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dkernel_test.cc122 // Add_2 Add_3 in TEST_F()
163 // Add_2 Add_3 in TEST_F()
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_mkldnn_pattern_matcher.py1301 # [qconv2d_pointwise_default_1, dequantize_per_tensor, add_3, quantize_per_tensor]
1360 … # [qconv2d_pointwise_default_1, dequantize_per_tensor, add_3, relu, quantize_per_tensor]
1421 # [qconv2d_pointwise_default_1, add_3]
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_control_flow.py597 add_3 = torch.ops.aten.add.Tensor(mul_4, mul_3); mul_4 = mul_3 = None
598 mul_5 = torch.ops.aten.mul.Tensor(add_3, add)
599 mul_6 = torch.ops.aten.mul.Tensor(add_3, add); add_3 = add = None
H A Dtest_aotdispatch.py3453 def f(self_s_emb, add_3): argument
3454 einsum_2 = torch.functional.einsum("ah,th->t", self_s_emb, add_3)
4281 add_3 = torch.ops.aten.add.Tensor(cos, sin); cos = sin = None
4283 return (add_3,)""",
/aosp_15_r20/external/ComputeLibrary/examples/
H A Dgraph_edsr.h826 node_add_3->set_common_node_parameters(NodeParams{ "add_3", target }); in setup()
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_repros.py5792 add_3 = torch.ops.aten.add.Tensor(slice_5, slice_6)
5798 unsqueeze_1 = torch.ops.aten.unsqueeze.default(add_3, 1)
5799 add_3 = None
H A Dtest_export.py3935 add_3 = add_2 + cos_2; add_2 = cos_2 = None
3936 return (add_3,)""",
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer.cc529 // x y w Add_3
541 // x y w Add_3 AddN(x, y, q, e) z
/aosp_15_r20/external/vixl/test/aarch64/
H A Dtest-cpu-features-aarch64.cc180 TEST_NONE(add_3, add(w0, w1, 0xb7e << 12))
786 TEST_NEON(add_3, add(v0.V8H(), v1.V8H(), v2.V8H()))

12