Home
last modified time | relevance | path

Searched defs:input_features (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/test/optim/
H A Dtest_swa_utils.py28 def __init__(self, input_features): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_pad_for_convolutions.cc228 auto input_features = in TryResolvePaddedShapesForTensorCore() local
H A Dgpu_conv_rewriter.cc510 int64_t input_features = rhs->shape().dimensions(input_feature_dimension); in MatchBackwardInput() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference.cc1739 const int64_t input_features = in InferConvolveShape() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dshape_inference.cc1664 auto input_features = input_tensor_shape[dnums.input_feature_dimension()]; in PrecheckForXlaConvV2Op() local