HomeSort by relevance Sort by last modified time
    Searched refs:tensor_offset (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
quantized_mul_op.cc 192 int32 tensor_offset, int64 tensor_num_elements,
197 (static_cast<int32>(tensor_data[i]) - tensor_offset);
205 const quint8* tensor_data, int32 tensor_offset, int64 tensor_num_elements,
208 const uint8x8_t offset_y_8x8 = vmov_n_u8(tensor_offset);
272 (static_cast<int32>(tensor_data[i]) - tensor_offset);
335 int32 tensor_offset; variable
342 tensor_offset = offset_y;
349 tensor_offset = offset_x;
353 tensor_offset, tensor_num_elements, z_data);
depthwise_conv_op_gpu.cu.cc 209 const int tensor_offset =
260 tile_ptr[tile_offset] = ldg(tensor_offset + in_ptr);
295 out_ptr[tensor_offset] = sum2;
    [all...]

Completed in 75 milliseconds