Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_pad_for_tensor_cores.cc166 auto input_features = in PadForTensorCores() local
Dcudnn_conv_rewriter.cc480 int64 input_features = new_shape.dimensions(input_feature_dimension); in MatchBackwardInput() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc1698 const int64 input_features = lhs.dimensions(dnums.input_feature_dimension()); in InferConvolveShape() local