Searched refs:DIMS (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | where_op.cc | 84 template <int DIMS, typename T, typename TIndex>
85 struct Where<CPUDevice, DIMS, T, TIndex> {
88 const typename Eigen::DSizes<TIndex, DIMS>& strides, TIndex true_n, in WriteIndexRowMajor()
90 for (int i = 0; i < DIMS; ++i) { in WriteIndexRowMajor()
98 typename TTypes<T, DIMS>::ConstTensor input, in Compute()
100 Eigen::DSizes<Eigen::DenseIndex, DIMS> dims = input.dimensions(); in Compute()
101 Eigen::DSizes<TIndex, DIMS> strides; in Compute()
107 strides[DIMS - 1] = 1; in Compute()
108 for (int i = DIMS - 2; i >= 0; --i) { in Compute()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_test.cc | 60 #define DIMS(n, h, w, c) \ macro
72 #define DIMS(n, h, w, c) \ macro
94 DIMS(batch_size, input_height, input_width, input_depth)); in SimpleConv2D()
108 filter, DIMS(1, stride, stride, 1), padding, in SimpleConv2D()
126 &input_data, DIMS(batch_size, input_height, input_width, input_depth)); in SimpleConv2DBackpropInput()
141 DIMS(batch_size, output_height, output_width, filter_count)); in SimpleConv2DBackpropInput()
153 attrs = attrs.Dilations(DIMS(1, 2, 2, 1)); in SimpleConv2DBackpropInput()
157 DIMS(1, stride, stride, 1), padding, attrs); in SimpleConv2DBackpropInput()
516 DIMS(kBatchSize, kHeight, kWidth, kDepthIn), in TEST_F()
|