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.h47 typedef enum { TENSORS_INTERLEAVED, TENSORS_CONTIGUOUS } MklTfTensorOrdering; typedef
50 static const MklTfTensorOrdering kTensorOrdering = TENSORS_CONTIGUOUS;
54 if (kTensorOrdering == MklTfTensorOrdering::TENSORS_INTERLEAVED) { in DataIndexToMetaDataIndex()
59 CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in DataIndexToMetaDataIndex()
66 if (kTensorOrdering == MklTfTensorOrdering::TENSORS_INTERLEAVED) { in GetTensorDataIndex()
69 CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in GetTensorDataIndex()
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_tfconversion_pass_test.cc130 if (kTensorOrdering == MklTfTensorOrdering::TENSORS_INTERLEAVED) { in TEST_F()
154 CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in TEST_F()
185 if (kTensorOrdering == MklTfTensorOrdering::TENSORS_INTERLEAVED) { in TEST_F()
213 CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in TEST_F()
Dmkl_layout_pass_test.cc213 CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); \
516 CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); \
575 DCHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); \
624 DCHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); \
681 DCHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); \
732 DCHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); \
773 DCHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); \
814 DCHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); \
2303 DCHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); \
2365 DCHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); \
[all …]
Dmkl_layout_pass.cc2220 CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in SetUpContiguousInputs()
2403 if (kTensorOrdering == MklTfTensorOrdering::TENSORS_INTERLEAVED) { in SetUpInputs()
2410 CHECK_EQ(kTensorOrdering, MklTfTensorOrdering::TENSORS_CONTIGUOUS); in SetUpInputs()