Searched refs:output_row_offset (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | im2col_utils.h | 53 const int output_row_offset = (buffer_id * single_buffer_length); in ExtractPatchIntoBufferColumn() local
55 output_row_offset + (h_offset * kwidth + w_offset) * in_depth; in ExtractPatchIntoBufferColumn()
70 memset(conv_buffer_data + output_row_offset, zero_byte, in ExtractPatchIntoBufferColumn()
107 output_row_offset + in ExtractPatchIntoBufferColumn()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 806 int num_cols, int output_row_offset, int output_col_offset, bool assign,
1088 int output_row_offset, int output_col_offset, bool assign, in ComputeOutputBlock() argument
1102 const DSizes begin(output_row_offset, output_col_offset); in ComputeOutputBlock()
1124 std::swap(output_row_offset, output_col_offset); in ComputeOutputBlock()
1129 memcpy(&(*output)(output_row_offset + i, output_col_offset), in ComputeOutputBlock()
|