Searched defs:axis_shape (Results 1 – 7 of 7) sorted by relevance
35 std::vector<int> axis_shape = {1}; in TEST_F() local54 std::vector<int> axis_shape = {1}; in TEST_F() local73 std::vector<int> axis_shape = {1}; in TEST_F() local92 std::vector<int> axis_shape = {1}; in TEST_F() local123 std::vector<int> axis_shape = {1}; in TEST_F() local154 std::vector<int> axis_shape = {1}; in TEST_F() local185 std::vector<int> axis_shape = {1}; in TEST_F() local208 std::vector<int> axis_shape = {1}; in TEST_F() local231 std::vector<int> axis_shape = {1}; in TEST_F() local254 std::vector<int> axis_shape = {1}; in TEST_F() local
67 auto axis_shape = ctx.Input(1)->GetTensorShape(); in Compute() local141 auto axis_shape = axis->GetTensorShape()->GetDimSizes(); in ComputeReverseV2() local
49 int64_t axis_shape = 0; in ArgminV2InferShape() local
38 …auto axis_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->GetSha… in CumulativeLogsumexpInferShape() local
52 …auto axis_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape())[kShap… in ConcatOffsetV1InferShape() local
364 auto axis_shape = CheckAndConvertUtils::GetTensorInputShape(op_name, input_args, 1); in CheckAndGetAxisValueFromTensor() local494 int64_t axis_shape = 1; in ReduceBaseInferShape() local
91 int64_t axis_shape = 1; in ReduceFusionInferShape() local