Searched refs:MakeRandomTensor (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | conv_ops_test.cc | 1097 static Tensor MakeRandomTensor(const TensorShape& shape) { in MakeRandomTensor() function 1108 Tensor images_t = MakeRandomTensor({batch, height, width, in_depth}); in Conv2D() 1109 Tensor filter_t = MakeRandomTensor({filter_w, filter_h, in_depth, out_depth}); in Conv2D() 1136 Tensor bias_t = MakeRandomTensor({out_depth}); in Conv2DWithBias() 1183 Tensor scale_t = MakeRandomTensor({out_depth}); in Conv2DWithBatchNorm() 1184 Tensor offset_t = MakeRandomTensor({out_depth}); in Conv2DWithBatchNorm() 1185 Tensor mean_t = MakeRandomTensor({out_depth}); in Conv2DWithBatchNorm() 1186 Tensor variance_t = MakeRandomTensor({out_depth}); in Conv2DWithBatchNorm() 1236 Tensor images_t = MakeRandomTensor({batch, height, width, in_depth}); 1237 Tensor filter_t = MakeRandomTensor({filter_w, filter_h, in_depth, out_depth}); [all …]
|