Searched refs:slice_tensor (Results 1 – 2 of 2) sorted by relevance
452 const Tensor& slice_tensor) { in AddSlice() argument458 return Add(full_tensor_key, slice_tensor); in AddSlice()469 CHECK_EQ(full_entry->dtype(), slice_tensor.dtype()); in AddSlice()477 full_entry->set_dtype(slice_tensor.dtype()); in AddSlice()486 status_ = Add(slice_name, slice_tensor); in AddSlice()
141 const TensorSlice& slice_spec, const Tensor& slice_tensor);