Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.cc560 std::array<std::vector<int64>, 3> einsum_config_numeric; in ParseEinsumString() local
622 /*ellipsis_rank=*/0, &einsum_config_numeric[0])); in ParseEinsumString()
634 /*ellipsis_rank=*/0, &einsum_config_numeric[1])); in ParseEinsumString()
645 &einsum_config_numeric[2])); in ParseEinsumString()
646 return einsum_config_numeric; in ParseEinsumString()
684 auto einsum_config_numeric, in Einsum()
686 return Einsum(x, einsum_config_numeric[0], y, einsum_config_numeric[1], in Einsum()
687 einsum_config_numeric[2], precision); in Einsum()