Searched refs:output_row_offset (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 807 int num_cols, int output_row_offset, int output_col_offset, bool assign,
1089 int output_row_offset, int output_col_offset, bool assign, in ComputeOutputBlock() argument
1103 const DSizes begin(output_row_offset, output_col_offset); in ComputeOutputBlock()
1125 std::swap(output_row_offset, output_col_offset); in ComputeOutputBlock()
1130 memcpy(&(*output)(output_row_offset + i, output_col_offset), in ComputeOutputBlock()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1864 const int output_row_offset = (buffer_id * single_buffer_length); in ExtractPatchIntoBufferColumn() local
1866 output_row_offset + (h_offset * kwidth + w_offset) * in_depth; in ExtractPatchIntoBufferColumn()
1881 memset(conv_buffer_data + output_row_offset, zero_byte, in ExtractPatchIntoBufferColumn()
1918 output_row_offset + in ExtractPatchIntoBufferColumn()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 2001 const int output_row_offset = (buffer_id * single_buffer_length); in ExtractPatchIntoBufferColumn() local
2003 output_row_offset + (h_offset * kwidth + w_offset) * in_depth; in ExtractPatchIntoBufferColumn()
2018 memset(conv_buffer_data + output_row_offset, zero_byte, in ExtractPatchIntoBufferColumn()
2055 output_row_offset + in ExtractPatchIntoBufferColumn()
|