Home
last modified time | relevance | path

Searched refs:tensor_offset (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dquantized_mul_op.cc191 int32 tensor_offset, int64 tensor_num_elements, in VectorTensorMultiply() argument
196 (static_cast<int32>(tensor_data[i]) - tensor_offset); in VectorTensorMultiply()
204 const quint8* tensor_data, int32 tensor_offset, int64 tensor_num_elements, in VectorTensorMultiply() argument
207 const uint8x8_t offset_y_8x8 = vmov_n_u8(tensor_offset); in VectorTensorMultiply()
271 (static_cast<int32>(tensor_data[i]) - tensor_offset); in VectorTensorMultiply()
334 int32 tensor_offset; in Compute() local
341 tensor_offset = offset_y; in Compute()
348 tensor_offset = offset_x; in Compute()
352 tensor_offset, tensor_num_elements, z_data); in Compute()
Ddepthwise_conv_op_gpu.h232 const int tensor_offset =
283 tile_ptr[tile_offset] = static_cast<S>(ldg(tensor_offset + in_ptr));
319 out_ptr[tensor_offset] = static_cast<T>(sum2);
1203 const int tensor_offset = block_height * in_row_size;
1257 tile_ptr[tile_offset] = static_cast<S>(ldg(tensor_offset + in_ptr));
1269 skip_second ? S() : static_cast<S>(ldg(tensor_offset + out_ptr));
/external/tensorflow/tensorflow/lite/micro/
Dtest_helpers.cc893 const Offset<Tensor> tensor_offset = CreateTensor( in Create1dFlatbufferTensor() local
897 builder->Finish(tensor_offset); in Create1dFlatbufferTensor()
916 const Offset<Tensor> tensor_offset = CreateTensor( in CreateQuantizedFlatbufferTensor() local
920 builder->Finish(tensor_offset); in CreateQuantizedFlatbufferTensor()
934 const Offset<Tensor> tensor_offset = CreateTensor( in CreateMissingQuantizationFlatbufferTensor() local
938 builder->Finish(tensor_offset); in CreateMissingQuantizationFlatbufferTensor()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc3129 const size_t tensor_offset = static_unpacked_data_.size(); in PrepareOpsToDelegate() local
3130 static_unpacked_data_.resize(tensor_offset + context->tensors[t].bytes); in PrepareOpsToDelegate()
3132 char* unpacked_data = static_unpacked_data_.data() + tensor_offset; in PrepareOpsToDelegate()
3220 static_unpacked_data_map_[t] = tensor_offset; in PrepareOpsToDelegate()