Searched defs:input_a_shape (Results 1 – 6 of 6) sorted by relevance
29 auto input_a_shape = input_a_shape_ptr->GetShapeVector(); in InferShape() local
40 Shape input_a_shape = inputs_shape_.at(0); in InferExpandShape() local89 Shape input_a_shape = input_shapes.at(0); in BaseCheckStrategy() local402 Shape input_a_shape = input_shapes.at(0); in InferOutputLayout() local
79 Shape input_a_shape = inputs_shape_.at(0); in CheckStrategy() local
543 Shape input_a_shape = inputs_shape_.at(0); in CheckStrategy() local
532 Shape input_a_shape = input_a_tensor_info.shape(); in GetBackwardCommCost() local829 Shape input_a_shape = input_a_tensor_info.shape(); in GetBackwardComputationCost() local866 Shape input_a_shape = input_a_tensor_info.shape(); in GetBackwardCommCost() local1354 Shape input_a_shape = input_a_tensor_info.shape(); in GetBackwardCommCost() local1613 Shape input_a_shape = input_a_tensor_info.shape(); in GetBackwardCommCost() local1924 Shape input_a_shape = input_a_tensor_info.shape(); in GetBackwardCommCost() local
1172 …def infer_shape(self, input_w1_shape, input_w2_shape, input_v_shape, input_a_shape, input_da_shape… argument