Searched refs:diff_shift_data (Results 1 – 1 of 1) sorted by relevance
1287 auto diff_shift_data = diff_shift_tensor->flat<U>().data(); in Compute() local1291 std::memcpy(reinterpret_cast<char*>(diff_shift_data), in Compute()1366 auto diff_shift_data = (*diff_shift_tensor)->flat<U>().data(); in AllocateTFOutputs() local1367 std::fill_n(diff_shift_data, (*diff_shift_tensor)->shape().num_elements(), in AllocateTFOutputs()