Searched refs:input_shape_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fft_thunk.cc | 106 input_shape_(input_shape), in FftThunk()
113 VLOG(3) << "Input shape: " << ShapeUtil::HumanStringWithLayout(input_shape_); in ExecuteOnStream()
125 for (int i = 0; i < input_shape_.dimensions_size() - fft_rank; ++i) { in ExecuteOnStream()
126 batch_size *= input_shape_.dimensions(i); in ExecuteOnStream()
137 auto dim_offset = input_shape_.dimensions_size() - fft_rank + i; in ExecuteOnStream()
139 input_embed[i] = input_shape_.dimensions(dim_offset); in ExecuteOnStream()
140 input_distance *= input_shape_.dimensions(dim_offset); in ExecuteOnStream()
|
| D | fft_thunk.h | 90 const Shape input_shape_; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | winograd_transform.h | 34 : filter_shape_(3, 3), input_shape_(4, 4), output_shape_(2, 2) {} in WinogradTransform()
46 virtual const Shape& input_shape() const { return input_shape_; } in input_shape()
51 const Shape input_shape_; variable
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1045 : input_shape_(input_shape), in FullyConnectedAsGEMVWorkerTask()
1065 input_shape_, input_data_, input_offset_, filter_shape_, filter_data_, in Run()
1072 const RuntimeShape& input_shape_; member
1885 input_shape_(input_shape), in MeanWorkerTask()
1897 MeanImpl(op_params_, input_shape_, input_data_, input_zero_point_, in Run()
1904 const RuntimeShape& input_shape_; member
|