Home
last modified time | relevance | path

Searched defs:weights_buffer (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dconvert_pure_conv_to_depthwise.cc99 auto& weights_buffer = in Run() local
H A Dfuse_binary_into_following_affine.cc72 const auto& weights_buffer = weights.GetBuffer<ArrayDataType::kFloat>(); in FuseAddOrSubParamsIntoFollowingAffine() local
H A Dfuse_binary_into_preceding_affine.cc129 auto& weights_buffer = weights.GetMutableBuffer<ArrayDataType::kFloat>(); in FuseMulOrDivParamsIntoPrecedingAffine() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantize_model_test.cc685 const auto weights_buffer = model_.buffers[weights_tensor->buffer].get(); in TEST_F() local
803 const auto weights_buffer = model_.buffers[weights_tensor->buffer].get(); in TEST_P() local
908 const auto weights_buffer = model_.buffers[weights_tensor->buffer].get(); in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
H A Dquantize_model_test.cc495 const auto weights_buffer = model_.buffers[weights_tensor->buffer].get(); in TEST_P() local
602 const auto weights_buffer = model_.buffers[weights_tensor->buffer].get(); in TEST_P() local