Home
last modified time | relevance | path

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

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