Searched refs:kInputIndex3 (Results 1 – 14 of 14) sorted by relevance
67 if (dout_shape_norm[kInputIndex3] != abstract::Shape::SHP_ANY && in SetPadList()68 x_size_v[kInputIndex3] != abstract::Shape::SHP_ANY) { in SetPadList()70 …(dout_shape_norm[kInputIndex3] - 1) * stride_w + dilation_w * (kernel_w - 1) + 1 - x_size_v[kInput… in SetPadList()
63 auto reserve_1 = input_args[kInputIndex3]->Broaden(); in BatchNormGradInfer()
46 if (input_args[kInputIndex3]->BuildType() != nullptr) { in BinaryCrossEntroyGradInferType()
86 …auto mean = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex3]->BuildShape()… in BatchNormInfer()122 (void)args_moving.emplace("bias", input_args[kInputIndex3]->BuildType()); in BatchNormInfer()
37 …auto dense_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex3]->Build… in SparseToDenseInfer()
77 auto g_type = input_args[kInputIndex3]->BuildType(); in ApplyMomentumInfer()
64 if (input_args[kInputIndex3]->BuildType() != nullptr) { in BinaryCrossEntroyInferType()
82 {"off_dtype", input_args[kInputIndex3]->BuildType()}}; in OneHotInferType()
99 …(void)CheckAndConvertUtils::CheckTensorTypeValid("sequence_length", input_args[kInputIndex3]->Buil… in InferType()
70 auto grad_type = input_args[kInputIndex3]->BuildType(); in InferType()
94 auto infer_type3 = input_args[kInputIndex3]->BuildType()->cast<TensorTypePtr>()->element(); in LstmInfer()
265 kInputIndex3, enumerator
211 auto strides_v = CheckAndGetValidStrides(input_args[kInputIndex3]); in StridedSliceInferShape()
134 …cast<int64_t>(std::floor(1 + ((x_w * 1) + pad_list->at(kInputIndex2) + pad_list->at(kInputIndex3) - in Conv2DPadFunction()