Searched defs:input_a_shape (Results 1 – 4 of 4) sorted by relevance
38 Shape input_a_shape = inputs_shape_.at(0); in InferExpendShape() local84 Shape input_a_shape = input_shapes.at(0); in CheckStrategy() local
80 Shape input_a_shape = inputs_shape_.at(0); in CheckStrategy() local
453 Shape input_a_shape = input_a_tensor_info.shape(); in GetBackwardCommCost() local750 Shape input_a_shape = input_a_tensor_info.shape(); in GetBackwardComputationCost() local785 Shape input_a_shape = input_a_tensor_info.shape(); in GetBackwardCommCost() local1270 Shape input_a_shape = input_a_tensor_info.shape(); in GetBackwardCommCost() local1529 Shape input_a_shape = input_a_tensor_info.shape(); in GetBackwardCommCost() local
1326 …def infer_shape(self, input_w1_shape, input_w2_shape, input_v_shape, input_a_shape, input_da_shape… argument