Searched defs:tensor_slices (Results 1 – 4 of 4) sorted by relevance
108 const int tensor_slices) in MeanStdDevNormalization()251 const int tensor_slices) { in CreateMeanStdDevNormalization()
115 std::unordered_map<string, TensorSliceSet*>* tensor_slices) { in RegisterTensorSlice()
173 std::vector<TensorSlice> tensor_slices; in Compute() local
44 def _save(filename, tensor_names, tensors, tensor_slices=None, name="save"): argument