Searched defs:tensor_slices (Results 1 – 4 of 4) sorted by relevance
115 std::unordered_map<string, TensorSliceSet*>* tensor_slices) { in RegisterTensorSlice()
178 std::vector<TensorSlice> tensor_slices; in Compute() local
39 def _save(filename, tensor_names, tensors, tensor_slices=None, name="save"): argument
138 const int tensor_slices = DivideRoundUp(shape.c, 4); in MeanStdDevNormalization() local