| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gemm_thunk.h | 71 const Shape output_shape_; member in class:xla::gpu::GemmThunk
|
| fft_thunk.cc | 124 output_shape_(output_shape) {}
131 << ShapeUtil::HumanStringWithLayout(output_shape_);
156 output_embed[i] = output_shape_.dimensions(dim_offset);
157 output_distance *= output_shape_.dimensions(dim_offset);
193 ->ThenBlasScal(ShapeUtil::ElementsIn(output_shape_),
217 ->ThenBlasScal(ShapeUtil::ElementsIn(output_shape_),
|
| gemm_thunk.cc | 222 output_shape_(output_shape),
241 int64 output_num_rows = output_shape_.dimensions(0);
242 int64 output_num_cols = output_shape_.dimensions(1);
269 LayoutUtil::Minor(output_shape_.layout(), 0);
284 PrimitiveType element_type = output_shape_.element_type();
323 if (LayoutUtil::Minor(output_shape_.layout(), 0) == 0) {
|
| convolution_thunk.cc | 58 output_shape_(output_shape),
79 convolution_kind_, input_shape_, filter_shape_, output_shape_, input_data,
|
| fft_thunk.h | 92 const Shape output_shape_; member in class:xla::gpu::FftThunk
|
| convolution_thunk.h | 97 const Shape output_shape_; member in class:xla::gpu::ConvolutionThunk
|
| /external/tensorflow/tensorflow/python/ops/ |
| nn_ops.py | [all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| winograd_transform.h | 34 : filter_shape_(3, 3), input_shape_(4, 4), output_shape_(2, 2) {}
47 virtual const Shape& output_shape() const { return output_shape_; }
52 const Shape output_shape_; member in class:tensorflow::WinogradTransform
|
