Searched refs:depth_in_ (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | virtual_scheduler_test.cc | 700 s.WithOpName("x"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithConv2Ds() 702 s.WithOpName("y"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithConv2Ds() 704 s.WithOpName("z"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithConv2Ds() 706 s.WithOpName("f"), {kernel_, kernel_, depth_in_, depth_out_}, DT_FLOAT); in CreateGrapplerItemWithConv2Ds() 725 s.WithOpName("x"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithConv2DAndVariable() 727 {kernel_, kernel_, depth_in_, depth_out_}, DT_FLOAT); in CreateGrapplerItemWithConv2DAndVariable() 830 s.WithOpName("x"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithAddFromOneTensor() 853 s.WithOpName("x"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithSwitchMergeInput() 857 s.WithOpName("b"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithSwitchMergeInput() 861 s.WithOpName("z"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithSwitchMergeInput() [all …]
|