Home
last modified time | relevance | path

Searched defs:vector_dim (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgemm_rewriter.cc338 int64_t vector_dim = it - broadcast_dims.begin(); in FuseVectorBiasAdd() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc3457 int vector_dim = -1; in GetTensorVectorSizeAndDim() local
3473 int vector_dim = -1; in GetTensorVectorSizeAndDim() local
3516 int vector_size, vector_dim; in GetCudnnOperationGraph() local
3651 int vector_size, vector_dim; in GetCudnnFusedOperationGraph() local