Searched refs:diff_shift_tensor (Results 1 – 1 of 1) sorted by relevance
1281 Tensor* diff_shift_tensor = nullptr; in Compute() local1283 &diff_shift_tensor); in Compute()1287 auto diff_shift_data = diff_shift_tensor->flat<U>().data(); in Compute()1330 Tensor* diff_shift_tensor = nullptr; in HandleEmptyInput() local1332 &diff_shift_tensor); in HandleEmptyInput()1338 Tensor** diff_shift_tensor) { in AllocateTFOutputs() argument1340 DCHECK(diff_shift_tensor); in AllocateTFOutputs()1361 AllocateOutputSetMklShape(context, kDiffShiftIndex, diff_shift_tensor, in AllocateTFOutputs()1364 DCHECK(*diff_shift_tensor); in AllocateTFOutputs()1366 auto diff_shift_data = (*diff_shift_tensor)->flat<U>().data(); in AllocateTFOutputs()[all …]