Home
last modified time | relevance | path

Searched refs:leading_dimension (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_utils.h63 int64 leading_dimension,
Dtensor_list_utils.cc192 int64 leading_dimension, in BuildUninitializedTensorList() argument
197 auto broadcast = xla::Broadcast(zero, std::vector<int64>{leading_dimension}); in BuildUninitializedTensorList()