Home
last modified time | relevance | path

Searched refs:slice_tensor (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc452 const Tensor& slice_tensor) { in AddSlice() argument
458 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()
Dtensor_bundle.h141 const TensorSlice& slice_spec, const Tensor& slice_tensor);