Searched refs:MklTfTensorOrdering (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/graph/ |
D | mkl_graph_util.h | 41 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()
|
D | mkl_layout_pass_test.cc | 181 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 …]
|
D | mkl_tfconversion_pass_test.cc | 131 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()
|
D | mkl_layout_pass.cc | 1710 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()
|