Searched defs:tensor_slices (Results 1 – 3 of 3) sorted by relevance
174 std::unordered_map<string, TensorSliceSet*>* tensor_slices) { in RegisterTensorSlice()
169 std::vector<TensorSlice> tensor_slices; in Compute() local
42 def _save(filename, tensor_names, tensors, tensor_slices=None, name="save"): argument