Searched defs:output_shape_ (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ | ||
| D | fft_thunk.h | 91 const Shape output_shape_; variable |
| D | gemm_thunk.h | 71 const Shape output_shape_; variable |
| /external/tensorflow/tensorflow/core/kernels/ | ||
| D | winograd_transform.h | 52 const Shape output_shape_; variable |
| /external/tensorflow/tensorflow/lite/kernels/ | ||
| D | sparse_to_dense_test.cc | 59 int output_shape_; member in tflite::__anona42cd54f0111::SparseToDenseOpModel |
| D | transpose_conv_test.cc | 74 int output_shape_; member in tflite::__anon558e5be50111::TransposeConvOpModel |
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ | ||
| D | optimized_ops.h | 1085 const RuntimeShape& output_shape_; member |