Searched refs:SetTfDimOrder (Results 1 – 2 of 2) sorted by relevance
456 SetTfDimOrder(dims, format); in SetTfLayout()489 inline void SetTfDimOrder(const size_t dimension, const mkldnn_dims_t map) { in SetTfDimOrder() function496 inline void SetTfDimOrder(const size_t dimension, TensorFormat data_format) { in SetTfDimOrder() function519 inline void SetTfDimOrder(const size_t dimension, MklTensorFormat format) { in SetTfDimOrder() function521 SetTfDimOrder(dimension, data_format); in SetTfDimOrder()
1251 dnn_shape_diff_src.SetTfDimOrder(src_dims.size(), tensor_format_); in Compute()