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