Searched refs:einsum_config_numeric (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 596 std::array<std::vector<int64_t>, 3> einsum_config_numeric; in ParseEinsumString() local
658 /*ellipsis_rank=*/0, &einsum_config_numeric[0])); in ParseEinsumString()
670 /*ellipsis_rank=*/0, &einsum_config_numeric[1])); in ParseEinsumString()
681 &einsum_config_numeric[2])); in ParseEinsumString()
682 return einsum_config_numeric; in ParseEinsumString()
721 auto einsum_config_numeric, in Einsum()
723 return Einsum(x, einsum_config_numeric[0], y, einsum_config_numeric[1], in Einsum()
724 einsum_config_numeric[2], precision, preferred_element_type); in Einsum()
|