Searched refs:unaligned_flat (Results 1 – 13 of 13) sorted by relevance
37 const auto s1 = t1.unaligned_flat<tstring>();38 const auto s2 = t2.unaligned_flat<tstring>();
105 const T* Tx = x.unaligned_flat<T>().data(); in ExpectEqual()106 const T* Ty = y.unaligned_flat<T>().data(); in ExpectEqual()155 const T* Tx = x.unaligned_flat<T>().data(); in ExpectClose()156 const T* Ty = y.unaligned_flat<T>().data(); in ExpectClose()
52 output->unaligned_flat<tstring>() = input.unaligned_flat<tstring>(); in DeepCopy()55 output->unaligned_flat<Variant>() = input.unaligned_flat<Variant>(); in DeepCopy()
147 EXPECT_EQ(2, y.unaligned_flat<int32>()(i)); in TEST()177 EXPECT_EQ("goodbye", y.unaligned_flat<tstring>()(i)); in TEST()210 y.unaligned_flat<Variant>()(i).get<Tensor>()->scalar<tstring>()()); in TEST()
1296 y.unaligned_flat<float>().setConstant(1.0); in TEST()1298 EXPECT_EQ(1.0, y.unaligned_flat<float>()(i)); in TEST()1322 auto tz = z.unaligned_flat<float>(); in TEST()1335 y.unaligned_flat<float>().setConstant(1.0); in TEST()1337 EXPECT_EQ(1.0, y.unaligned_flat<float>()(i)); in TEST()
465 typename TTypes<T>::UnalignedFlat unaligned_flat() { in unaligned_flat() function554 typename TTypes<T>::UnalignedConstFlat unaligned_flat() const { in unaligned_flat() function
525 col_params->group.group_size = group_size.unaligned_flat<int32>()(0); in ComputeAsync()526 col_params->group.group_key = group_key.unaligned_flat<int32>()(0); in ComputeAsync()528 col_params->instance.instance_key = instance_key.unaligned_flat<int32>()(0); in ComputeAsync()660 col_params->group.group_size = group_size.unaligned_flat<int32>()(0); in ComputeAsync()665 col_params->group.group_key = group_key.unaligned_flat<int32>()(0); in ComputeAsync()667 col_params->instance.instance_key = instance_key.unaligned_flat<int32>()(0); in ComputeAsync()788 col_params->group.group_size = group_size.unaligned_flat<int32>()(0); in ComputeAsync()789 col_params->group.group_key = group_key.unaligned_flat<int32>()(0); in ComputeAsync()791 col_params->instance.instance_key = instance_key.unaligned_flat<int32>()(0); in ComputeAsync()922 col_params->group.group_size = group_size.unaligned_flat<int32>()(0); in ComputeAsync()[all …]
536 tmp.unaligned_flat<T>(); in Compute()685 tmp.unaligned_flat<T>(); in Compute()711 tmp.unaligned_flat<T>(); in Scatter()
182 const auto& fanout_tensor_data = fanout_tensor.unaligned_flat<int32>(); in IsCancellableConstPermTransposeNodePair()183 const auto& fanin_tensor_data = fanin_tensor.unaligned_flat<int32>(); in IsCancellableConstPermTransposeNodePair()
1060 const auto& tensor_data = tensor.unaligned_flat<int32>(); in IsValidConstPermTransposeNode()
772 original.unaligned_flat<T>()); in FinishTimeStep()810 functor::TensorCopyUnaligned<Device, T>()(device_, t.unaligned_flat<T>(), in AlignTensor()1008 cs_tensor.unaligned_flat<T>()); in Compute()1010 h_tensor.unaligned_flat<T>()); in Compute()1294 device, x_grad_tensor.unaligned_flat<T>()); in Compute()
327 auto alias = actual.template unaligned_flat<T>(); in RunTest()
349 auto alias = actual.template unaligned_flat<T>(); in RunTest()