Home
last modified time | relevance | path

Searched refs:depth_in_ (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler_test.cc700 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 …]