Searched refs:new_strides (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | conv_ops_3d_test.py | 527 new_strides = test_util.NHWCToNCHW(strides)
530 new_strides = strides
535 new_strides,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 2805 std::vector<int32> new_strides; in CopyFormatAttrsConv() local
2810 new_strides = {strides[NDHWC::dim::N], strides[NDHWC::dim::C], in CopyFormatAttrsConv()
2821 new_strides = {strides[NHWC::dim::N], strides[NHWC::dim::C], in CopyFormatAttrsConv()
2827 nb->Attr("strides", new_strides); in CopyFormatAttrsConv()
2857 std::vector<int32> new_strides; in CopyAttrsPooling() local
2863 new_strides = {strides[NDHWC::dim::N], strides[NDHWC::dim::C], in CopyAttrsPooling()
2875 new_strides = {strides[NHWC::dim::N], strides[NHWC::dim::C], in CopyAttrsPooling()
2881 nb->Attr("strides", new_strides); in CopyAttrsPooling()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 4524 std::vector<int64_t> new_strides = slice->slice_strides(); in TryToReorderSliceAndReverse() local
4546 new_limits, new_strides)); in TryToReorderSliceAndReverse()
|