Searched refs:new_strides (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | conv_ops_3d_test.py | 520 new_strides = test_util.NHWCToNCHW(strides)
523 new_strides = strides
528 new_strides,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 2841 std::vector<int32> new_strides; in CopyFormatAttrsConv() local
2846 new_strides = {strides[NDHWC::dim::N], strides[NDHWC::dim::C], in CopyFormatAttrsConv()
2857 new_strides = {strides[NHWC::dim::N], strides[NHWC::dim::C], in CopyFormatAttrsConv()
2863 nb->Attr("strides", new_strides); in CopyFormatAttrsConv()
2893 std::vector<int32> new_strides; in CopyAttrsPooling() local
2899 new_strides = {strides[NDHWC::dim::N], strides[NDHWC::dim::C], in CopyAttrsPooling()
2911 new_strides = {strides[NHWC::dim::N], strides[NHWC::dim::C], in CopyAttrsPooling()
2917 nb->Attr("strides", new_strides); in CopyAttrsPooling()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 606 SmallVector<int64_t, 2> new_shape, new_strides, broadcast_dims; in InsertReshapeIfNecessary() local
622 new_strides.push_back(operand_strides[index]); in InsertReshapeIfNecessary()
635 new_strides.push_back(1); in InsertReshapeIfNecessary()
647 makeStridedLinearLayoutMap(new_strides, operand_offset, in InsertReshapeIfNecessary()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 4103 std::vector<int64> new_strides = slice->slice_strides(); in TryToReorderSliceAndReverse() local
4125 new_limits, new_strides)); in TryToReorderSliceAndReverse()
|