| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| convolution_thunk.cc | 57 filter_shape_(filter_shape),
79 convolution_kind_, input_shape_, filter_shape_, output_shape_, input_data,
|
| convolution_thunk.h | 96 const Shape filter_shape_; member in class:xla::gpu::ConvolutionThunk
|
| /external/tensorflow/tensorflow/core/kernels/ |
| winograd_transform.h | 34 : filter_shape_(3, 3), input_shape_(4, 4), output_shape_(2, 2) {}
45 virtual const Shape& filter_shape() const { return filter_shape_; }
50 const Shape filter_shape_; member in class:tensorflow::WinogradTransform
|
