OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Automated Code Change

Browse Source 
PiperOrigin-RevId: 847189651
This commit is contained in:
A. Unique TensorFlower
2025-12-20 13:45:28 -08:00
committed by TensorFlower Gardener
parent 8633fb9dcf
commit b1082f97e5
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/core/kernels/batching_util/concat_split_util.h
View File

@@ -81,7 +81,7 @@ absl::Status Concat(OpKernelContext* context,
(defined(TENSORFLOW_USE_ROCM) && TENSORFLOW_USE_ROCM)
if (std::is_same<Device, GPUDevice>::value) {
ConcatGPU<T>(context, inputs_flat, output, &output_flat);
return OkStatus();
return absl::OkStatus();
}
#endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
ConcatCPU<T>(context->device(), inputs_flat, &output_flat);
@@ -198,9 +198,9 @@ absl::Status SplitCPU(OpKernelContext* context, const Tensor& input,
// Handles the general case, on GPU.
template <typename T>
Status SplitGPU(OpKernelContext* context, const Tensor& input,
const gtl::ArraySlice<int64_t>& sizes,
std::vector<Tensor>* outputs) {
absl::Status SplitGPU(OpKernelContext* context, const Tensor& input,
const absl::Span<const int64_t>& sizes,
std::vector<Tensor>* outputs) {
// TODO(olston, apassos): Implement this.
LOG(FATAL) << "Not yet implemented"; // Crash ok
}