| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| fft_thunk.cc | 123 input_shape_(input_shape),
129 VLOG(3) << "Input shape: " << ShapeUtil::HumanStringWithLayout(input_shape_);
140 for (int i = 0; i < input_shape_.dimensions_size() - fft_rank; ++i) {
141 batch_size *= input_shape_.dimensions(i);
152 auto dim_offset = input_shape_.dimensions_size() - fft_rank + i;
154 input_embed[i] = input_shape_.dimensions(dim_offset);
155 input_distance *= input_shape_.dimensions(dim_offset);
|
| convolution_thunk.cc | 56 input_shape_(input_shape),
79 convolution_kind_, input_shape_, filter_shape_, output_shape_, input_data,
|
| fft_thunk.h | 91 const Shape input_shape_; member in class:xla::gpu::FftThunk
|
| convolution_thunk.h | 95 const Shape input_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) {}
46 virtual const Shape& input_shape() const { return input_shape_; }
51 const Shape input_shape_; member in class:tensorflow::WinogradTransform
|
