HomeSort by relevance Sort by last modified time
    Searched defs:output_shape_ (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gemm_thunk.h 71 const Shape output_shape_; member in class:xla::gpu::GemmThunk
convolution_thunk.h 97 const Shape output_shape_; member in class:xla::gpu::ConvolutionThunk
fft_thunk.h 92 const Shape output_shape_; member in class:xla::gpu::FftThunk
  /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

Completed in 80 milliseconds