Home
last modified time | relevance | path

Searched refs:tensor_c (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseSemiStructuredLinear.cu56 const Tensor& tensor_c, const Tensor& meta) {
222 TORCH_CHECK(tensor_c.dtype() == tensor_d_dtype,
224 tensor_d_dtype, ", but got ", tensor_c.dtype());
250 return {(ElementC*)tensor_c.data_ptr(),
327 const Tensor& tensor_a, const Tensor& tensor_b, const Tensor& tensor_c,
356 tensor_c,
378 tensor_c,
400 tensor_c,
422 tensor_c,
449 const Tensor& tensor_a, const Tensor& tensor_b, const Tensor& tensor_c,
[all …]
H A DSparseSemiStructuredOps.cu50 const Tensor& tensor_c, const Tensor& tensor_e, const Scalar& alpha,
235 return {(ElementC*)tensor_c.data_ptr(),
314 const Tensor& tensor_a, const Tensor& tensor_b, const Tensor& tensor_c,
343 tensor_c,
368 tensor_c,
393 tensor_c,
418 tensor_c,
447 const Tensor& tensor_a, const Tensor& tensor_b, const Tensor& tensor_c,
450 if (tensor_c.numel() > 0) {
466 tensor_c,
[all …]
/aosp_15_r20/external/federated-compute/fcp/aggregation/tensorflow/
H A Dcheckpoint_reader_test.cc46 auto tensor_c = CreateStringTfTensor({}, {"foobar"}); in TEST() local
48 {tensor_a, tensor_b, tensor_c}) in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/core/function/trace_type/
H A Dtrace_type_test.py124 tensor_c = array_ops.zeros(
130 self.assertNotEqual(tensor_a, tensor_c)
131 self.assertNotEqual(tensor_b, tensor_c)
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dtest_util_test.py767 tensor_c = array_ops.unique(values)[0]
769 self.assertShapeEqual(array_a, tensor_c)
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_common.py1350 tensor_c = torch.view_as_complex(tensor)
1358 dist.all_gather(tensor_list, tensor_c)
1360 dist.all_gather(tensor_list_c, tensor_c)