Searched refs:depth_in_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 119 s.WithOpName("x"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithConv2Ds()
121 s.WithOpName("y"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithConv2Ds()
123 s.WithOpName("z"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithConv2Ds()
125 s.WithOpName("f"), {kernel_, kernel_, depth_in_, depth_out_}, DT_FLOAT); in CreateGrapplerItemWithConv2Ds()
146 s.WithOpName("x"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithConv2DAndVariable()
148 {kernel_, kernel_, depth_in_, depth_out_}, DT_FLOAT); in CreateGrapplerItemWithConv2DAndVariable()
261 s.WithOpName("x"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithBatchNorm()
263 ops::RandomUniform(s.WithOpName("scale"), {depth_in_}, DT_FLOAT); in CreateGrapplerItemWithBatchNorm()
265 ops::RandomUniform(s.WithOpName("offset"), {depth_in_}, DT_FLOAT); in CreateGrapplerItemWithBatchNorm()
1549 s.WithOpName("x"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithInterDeviceTransfers()
[all …]
|