Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dmkl_graph_util.h41 typedef enum { TENSORS_INTERLEAVED, TENSORS_CONTIGUOUS } MklTfTensorOrdering; typedef
44 static const MklTfTensorOrdering kTensorOrdering = TENSORS_CONTIGUOUS;
48 if (kTensorOrdering == MklTfTensorOrdering::TENSORS_INTERLEAVED) { in DataIndexToMetaDataIndex()
53 CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in DataIndexToMetaDataIndex()
60 if (kTensorOrdering == MklTfTensorOrdering::TENSORS_INTERLEAVED) { in GetTensorDataIndex()
63 CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in GetTensorDataIndex()
Dmkl_layout_pass_test.cc181 CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in TEST_F()
440 CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in TEST_F()
491 DCHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in TEST_F()
534 DCHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in TEST_F()
586 DCHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in TEST_F()
631 DCHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in TEST_F()
667 DCHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in TEST_F()
697 DCHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in TEST_F()
1460 DCHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in TEST_F()
1516 DCHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in TEST_F()
[all …]
Dmkl_tfconversion_pass_test.cc131 if (kTensorOrdering == MklTfTensorOrdering::TENSORS_INTERLEAVED) { in TEST_F()
154 CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in TEST_F()
184 if (kTensorOrdering == MklTfTensorOrdering::TENSORS_INTERLEAVED) { in TEST_F()
211 CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in TEST_F()
Dmkl_layout_pass.cc1710 CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in SetUpContiguousInputs()
1882 if (kTensorOrdering == MklTfTensorOrdering::TENSORS_INTERLEAVED) { in SetUpInputs()
1889 CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in SetUpInputs()