Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_rewriter.cc338 int64_t vector_dim = it - broadcast_dims.begin(); in FuseVectorBiasAdd() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
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