Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc1703 std::vector<int64> expected_dnums(num_dims); in InferConvolveShape() local
1704 std::iota(expected_dnums.begin(), expected_dnums.end(), 0); in InferConvolveShape()
1715 if (input_dnums != expected_dnums) { in InferConvolveShape()
1721 if (window_dnums != expected_dnums) { in InferConvolveShape()
1727 if (output_dnums != expected_dnums) { in InferConvolveShape()