Searched refs:slice_tensor (Results 1 – 2 of 2) sorted by relevance
142 const TensorSlice& slice_spec, const Tensor& slice_tensor);
481 const Tensor& slice_tensor) { in AddSlice() argument487 return Add(full_tensor_key, slice_tensor); in AddSlice()498 CHECK_EQ(full_entry->dtype(), slice_tensor.dtype()); in AddSlice()506 full_entry->set_dtype(slice_tensor.dtype()); in AddSlice()515 status_ = Add(slice_name, slice_tensor); in AddSlice()