Searched refs:bias_buffer (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_lstm_merge_inputs.cc | 125 auto& bias_buffer = bias_array.GetMutableBuffer<ArrayDataType::kFloat>(); in Run() local
126 bias_buffer.data.resize(weights_dim1); in Run()
129 CopyArrayToSubArray(bias_buffer, weights_dim2, in Run()
132 CopyArrayToSubArray(bias_buffer, weights_dim2, in Run()
135 CopyArrayToSubArray(bias_buffer, weights_dim2, in Run()
138 CopyArrayToSubArray(bias_buffer, weights_dim2, in Run()
|
| D | ensure_bias_vectors.cc | 71 auto& bias_buffer = bias_array.GetMutableBuffer<ArrayDataType::kFloat>(); in ProcessLinearOperator() local
72 bias_buffer.data.resize(depth, 0.f); in ProcessLinearOperator()
|
| D | fuse_binary_into_preceding_affine.cc | 59 auto& bias_buffer = bias.GetMutableBuffer<ArrayDataType::kFloat>(); in FuseAddOrSubParamsIntoPrecedingAffine() local
60 float* const bias_data = bias_buffer.data.data(); in FuseAddOrSubParamsIntoPrecedingAffine()
131 auto& bias_buffer = bias.GetMutableBuffer<ArrayDataType::kFloat>(); in FuseMulOrDivParamsIntoPrecedingAffine() local
132 float* const bias_data = bias_buffer.data.data(); in FuseMulOrDivParamsIntoPrecedingAffine()
|
| D | fuse_binary_into_following_affine.cc | 74 auto& bias_buffer = bias.GetMutableBuffer<ArrayDataType::kFloat>(); in FuseAddOrSubParamsIntoFollowingAffine() local
75 float* const bias_data = bias_buffer.data.data(); in FuseAddOrSubParamsIntoFollowingAffine()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 173 se::DeviceMemoryBase bias_buffer, se::DeviceMemoryBase side_input_buffer, in LogFusedConvForwardAutotuneResults() argument
194 instr.set_bias_address(reinterpret_cast<uint64>(bias_buffer.opaque())); in LogFusedConvForwardAutotuneResults()
|
| D | gpu_utils.h | 233 se::DeviceMemoryBase bias_buffer, se::DeviceMemoryBase side_input_buffer,
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_model_test.cc | 653 const auto bias_buffer = model_.buffers[bias_tensor->buffer].get(); in TEST_F() local
654 ASSERT_EQ(bias_buffer->data.size(), sizeof(int32_t) * bias_tensor->shape[0]); in TEST_F()
656 reinterpret_cast<int32_t*>(bias_buffer->data.data()); in TEST_F()
762 const auto bias_buffer = model_.buffers[bias_tensor->buffer].get(); in TEST_P() local
767 ASSERT_EQ(bias_buffer->data.size(), control_size); in TEST_P()
774 int32_t* bias_values = reinterpret_cast<int32_t*>(bias_buffer->data.data()); in TEST_P()
780 int64_t* bias_values = reinterpret_cast<int64_t*>(bias_buffer->data.data()); in TEST_P()
|