Searched defs:new_tensor_layout (Results 1 – 2 of 2) sorted by relevance
1499 TensorLayout new_tensor_layout; in HandleCameAndAdaFactorOpt() local
608 auto new_tensor_layout = outputs_tensor_info_[kIndex0].tensor_layout(); in UpdateOutputTensorInfoForInterleaved() local