Searched refs:loss_shape (Results 1 – 5 of 5) sorted by relevance
78 abstract::ShapePtr loss_shape; in InferShape() local81 loss_shape = std::make_shared<abstract::Shape>(batch); in InferShape()83 …return std::make_shared<abstract::TupleShape>(std::vector<abstract::BaseShapePtr>{loss_shape, grad… in InferShape()88 loss_shape = std::make_shared<abstract::Shape>(batch, batch_min, batch_max); in InferShape()90 …return std::make_shared<abstract::TupleShape>(std::vector<abstract::BaseShapePtr>{loss_shape, grad… in InferShape()
39 std::vector<int64_t> loss_shape = {logits_shape[0]}; in SoftmaxCrossEntropyWithLogitsInfer() local49 auto output0 = std::make_shared<abstract::AbstractTensor>(logits_type, loss_shape); in SoftmaxCrossEntropyWithLogitsInfer()
124 std::vector<size_t> loss_shape; in CreateSoftmaxCrossEntropyWithLogits() local126 loss_shape.emplace_back(labels_shape[0]); in CreateSoftmaxCrossEntropyWithLogits()131 auto shapes = {loss_shape, AnfAlgo::GetOutputInferShape(one_hot_node, 0)}; in CreateSoftmaxCrossEntropyWithLogits()
2526 loss_shape = [logits_shape[0]]2528 return loss_shape, dlogits_shape2594 loss_shape = []2597 return loss_shape2836 loss_shape = []2837 return loss_shape
2369 auto loss_shape = loss_grad_layout.tensor_shape().array(); in SplitSens() local2370 if (loss_shape != sens_shape) { in SplitSens()2372 … << ShapeToString(sens_shape) << ", loss shape is " << ShapeToString(loss_shape); in SplitSens()