Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DMHA.cpp124 std::array<int, MAX_MHA_DIM> k_dim; member
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator.cc1069 double m_dim, n_dim, k_dim, k_dim_b = 0; in CountMatMulOperations() local
1197 TensorShapeProto_Dim k_dim; in GenerateBatchMatmulContextFromEinsum() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dnn_ops.cc1332 DimensionHandle k_dim; in TopKShapeFn() local