Searched refs:tensor_d (Results 1 – 4 of 4) sorted by relevance
51 const Scalar& beta, Tensor& tensor_d) {244 (ElementOutput*)tensor_d.data_ptr(),316 Tensor& tensor_d) {347 tensor_d);372 tensor_d);397 tensor_d);422 tensor_d);449 Tensor& tensor_d) {470 tensor_d);491 tensor_d);[all …]
228 Tensor tensor_d =259 (ElementOutput*)tensor_d.data_ptr(),308 return tensor_d;
370 for (int64_t tensor_d = oldshape.size() - 1; tensor_d >= 0; tensor_d--) { in computeStride_impl() local371 tensor_numel *= oldshape[tensor_d]; in computeStride_impl()373 if ((tensor_d == 0) || in computeStride_impl()374 (TORCH_GUARD_SIZE_OBLIVIOUS(sym_ne(oldshape[tensor_d - 1], 1)) && in computeStride_impl()375 oldstride[tensor_d - 1] != tensor_numel * chunk_base_stride)) { in computeStride_impl()385 if (tensor_d > 0) { in computeStride_impl()386 chunk_base_stride = oldstride[tensor_d - 1]; in computeStride_impl()
126 tensor_d = array_ops.ones([11, 3, 5],132 self.assertEqual(tensor_a, tensor_d)