Searched defs:from_layout (Results 1 – 7 of 7) sorted by relevance
66 TensorLayout from_layout; in TEST_F() local119 TensorLayout from_layout; in TEST_F() local171 TensorLayout from_layout; in TEST_F() local
36 auto from_layout = from.LayoutForRedistribution(); in MakeFromToLayout() local823 TensorLayout from_layout; in InferTensorRedistributionOperatorList() local940 Status TensorRedistribution::InferReshape(const TensorLayout &from_layout, const TensorLayout &to_l… in InferReshape()1031 Status TensorRedistribution::InferRedistribution(const TensorLayout &from_layout, const TensorLayou… in InferRedistribution()
85 TensorLayout from_layout() const { return this->from_; } in from_layout() function
174 TensorLayout from_layout; in TransformOperators() local
342 def _transform_tensor_by_layout(from_layout, to_layout, device_list, rank_id): argument
330 bool SpecialPatternInTransformer(const TensorLayout &from_layout, const TensorLayout &to_layout) { in SpecialPatternInTransformer()357 bool SkipTensorRedistribution(const TensorLayout &from_layout, const TensorLayout &to_layout) { in SkipTensorRedistribution()
359 TensorLayout from_layout = tensor_redistribution->layout_transfer().from_in(); in MatchingAccordingToIndex() local