| /external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| D | SparseSemiStructuredLinear.cu | 52 const Tensor& tensor_a,
183 const int length_m = tensor_a.size(0);
229 tensor_a.new_empty({length_m, length_n},
238 (ElementInputA*)tensor_a.data_ptr(), layout_a);
294 auto workspace = tensor_a.new_empty({(int64_t)workspace_size},
327 const Tensor& tensor_a, const Tensor& tensor_b, const Tensor& tensor_c,
330 const auto strides_a = tensor_a.strides();
352 tensor_a,
374 tensor_a,
396 tensor_a,
[all …]
|
| D | SparseSemiStructuredOps.cu | 48 const Tensor& tensor_a, const at::IntArrayRef::value_type& tensor_a_stride,
175 const int length_m = tensor_a.size(0);
203 (ElementInputA*)tensor_a.data_ptr(), layout_a);
284 auto workspace = tensor_a.new_empty({(int64_t)workspace_size},
314 const Tensor& tensor_a, const Tensor& tensor_b, const Tensor& tensor_c,
318 const auto strides_a = tensor_a.strides();
339 tensor_a,
364 tensor_a,
389 tensor_a,
414 tensor_a,
[all …]
|
| /external/tensorflow/tensorflow/lite/experimental/resource/ |
| D | resource_variable_test.cc | 98 TfLiteTensor tensor_a, tensor_b; in TEST() local
101 InitTensor(shape_a, kTfLiteDynamic, 1.0, &tensor_a); in TEST()
104 EXPECT_EQ(kTfLiteOk, var.AssignFrom(&tensor_a)); in TEST()
128 TfLiteTensorFree(&tensor_a); in TEST()
139 TfLiteTensor tensor_a, tensor_b; in TEST() local
142 InitTensor(shape_a, kTfLiteDynamic, 1.0, &tensor_a); in TEST()
145 EXPECT_EQ(kTfLiteOk, var.AssignFrom(&tensor_a)); in TEST()
169 TfLiteTensorFree(&tensor_a); in TEST()
|
| /external/federated-compute/fcp/aggregation/tensorflow/ |
| D | checkpoint_reader_test.cc | 42 auto tensor_a = in TEST() local
48 {tensor_a, tensor_b, tensor_c}) in TEST()
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | frozen_concat_linear.cpp | 143 bool isNonZeroDimEqual(Tensor& tensor_a, Tensor& tensor_b) { in isNonZeroDimEqual() argument
144 if (tensor_a.dim() != tensor_b.dim()) { in isNonZeroDimEqual()
147 for (int64_t i = 1; i < tensor_a.dim(); i++) { in isNonZeroDimEqual()
148 if (tensor_a.size(i) != tensor_b.size(i)) { in isNonZeroDimEqual()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | squared_difference.cc | 44 const auto& tensor_a = context->tensors[tensor_a_index]; in PopulateSubGraph() local
49 TF_LITE_ENSURE_STATUS(ComputeAndAddMinAndMax(context, tensor_a)); in PopulateSubGraph()
|
| /external/pytorch/benchmarks/transformer/ |
| D | sdp.py | 199 def assert_close_tensors(tensor_a, tensor_b): argument
201 if tensor_a.is_nested and tensor_b.is_nested:
202 for a, b in zip(tensor_a.unbind(), tensor_b.unbind()):
205 assert torch.allclose(tensor_a, tensor_b, atol=1e-3, rtol=1e-3)
|
| /external/pytorch/test/ |
| D | test_indexing.py | 1578 tensor_a = torch.rand(13, 11, 12, 13, 12).cpu()
1579 tensor_b = tensor_a.to(device=device)
1580 tensor_a[idx1] = 1.0
1581 tensor_a[idx1, :, idx2, idx2, :] = 2.0
1582 tensor_a[:, idx1, idx3, :, idx3] = 3.0
1586 self.assertEqual(tensor_a, tensor_b.cpu(), atol=0, rtol=0)
1588 tensor_a = torch.rand(10, 11).cpu()
1589 tensor_b = tensor_a.to(device=device)
1590 tensor_a[idx3] = 1.0
1591 tensor_a[idx2, :] = 2.0
[all …]
|
| /external/tensorflow/tensorflow/core/function/trace_type/ |
| D | trace_type_test.py | 120 tensor_a = array_ops.zeros([11, 3, 5],
129 self.assertNotEqual(tensor_a, tensor_b)
130 self.assertNotEqual(tensor_a, tensor_c)
132 self.assertEqual(tensor_a, tensor_d)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | benchmarks_test.py | 224 tensor_a = constant_op.constant(42.0)
226 self._benchmark_add(tensor_a, tensor_b)
229 tensor_a = constant_op.constant(42.0)
231 self._benchmark_add_operator_overload(tensor_a, tensor_b)
234 tensor_a = constant_op.constant(42)
236 self._benchmark_add(tensor_a, tensor_b)
239 tensor_a = constant_op.constant([[42.0, 42.0], [42.0, 42.0]])
241 self._benchmark_add(tensor_a, tensor_b)
244 tensor_a = constant_op.constant([[42, 42], [42, 42]])
246 self._benchmark_add(tensor_a, tensor_b)
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 4509 ITensorProxyPtr tensor_a = nullptr; in ConvertFullyConnectedImpl() local
4522 /*validation_only=*/false, &tensor_a, node_def, /*op_instance=*/0, in ConvertFullyConnectedImpl()
4525 VLOG(2) << "New shape of A " << DebugString(tensor_a->getDimensions()); in ConvertFullyConnectedImpl()
4563 *tensor_a->trt_tensor(), noutput, weights.GetTrtWeights(), in ConvertFullyConnectedImpl()
4624 ITensorProxyPtr tensor_a = convert_to_itensor(input_a); in ConvertMatMulImpl() local
4637 op_a = (tensor_a->getDimensions().nbDims < 2) in ConvertMatMulImpl()
4639 : get_matrix_op(tensor_a, transpose_a); in ConvertMatMulImpl()
4651 *tensor_a->trt_tensor(), op_a, *tensor_b->trt_tensor(), op_b); in ConvertMatMulImpl()
|