Searched refs:conv_config (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.h | 110 Status RunGpuConv(const GpuConvConfig& conv_config,
116 Status RunGpuConv(const GpuConvConfig& conv_config,
149 const GpuConvConfig& conv_config,
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 835 std::vector<int> conv_config(valid_conv_config); in TEST_F() local
836 conv_config[i] = 0; in TEST_F()
838 conv_config[0], conv_config[1], conv_config[2], conv_config[3], in TEST_F()
839 conv_config[4], conv_config[5], conv_config[6], conv_config[7]); in TEST_F()
|