| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/ |
| H A D | fully_connected.h | 36 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight()
82 const RuntimeShape& weights_shape, const int8_t* weights_data, in FullyConnectedSparseWeight1x16Impl()
118 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight1x4Impl()
189 const float* weights_data; member
202 const RuntimeShape& weights_shape, const int8_t* weights_data, in FullyConnectedSparseWeight1x16()
226 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight1x4()
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | fold_batch_norms_test.cc | 50 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldBatchNormsConv2D() local
101 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldBatchNormsDepthwiseConv2dNative() local
152 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldBatchNormsConv2DShared() local
210 Tensor weights_data(DT_FLOAT, TensorShape({2, 2})); in TestFoldBatchNormsMatMul() local
|
| H A D | fold_old_batch_norms_test.cc | 51 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldOldBatchNorms() local
135 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
213 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldFusedBatchNorms() local
290 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
469 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldFusedBatchNormsWithBatchToSpace() local
|
| H A D | fuse_convolutions_test.cc | 66 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFuseResizePadAndConv() local
116 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFuseResizeAndConv() local
168 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFusePadAndConv() local
|
| H A D | round_weights_test.cc | 49 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 10})); in TestRoundWeights() local
|
| H A D | flatten_atrous_test.cc | 64 Tensor weights_data(DT_FLOAT, TensorShape({2, 2, 2, 1})); in TestFlattenAtrousConv() local
|
| H A D | quantize_weights_test.cc | 50 Tensor weights_data(DT_FLOAT, weight_shape); in BuildGraphDef() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | unidirectional_sequence_rnn.mod.py | 20 activation, time_major, output, input_data, weights_data, argument
73 weights_data = [ variable
|
| D | bidirectional_sequence_rnn.mod.py | 113 weights_data = [ variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | unidirectional_sequence_rnn.mod.py | 20 activation, time_major, output, output_state, input_data, weights_data, argument
76 weights_data = [ variable
|
| D | bidirectional_sequence_rnn_1_3.mod.py | 119 weights_data = [ variable
|
| D | bidirectional_sequence_rnn_state_output.mod.py | 121 weights_data = [ variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | fuse_binary_into_following_affine.cc | 73 const float* const weights_data = weights_buffer.data.data(); in FuseAddOrSubParamsIntoFollowingAffine() local
137 float* weights_data = in FuseMulOrDivParamsIntoFollowingAffine() local
|
| H A D | shuffle_fc_weights.cc | 118 auto& weights_data = in Run() local
|
| H A D | fuse_binary_into_preceding_affine.cc | 130 float* const weights_data = weights_buffer.data.data(); in FuseMulOrDivParamsIntoPrecedingAffine() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_fused_batch_norm_op.cc | 85 void Execute(const T* src_data, const U* weights_data, T* dst_data, in Execute()
438 const T* diff_dst_data, const U* weights_data, T* diff_src_data, in Execute()
878 U* weights_data = reinterpret_cast<U*>(weights.GetAllocatedBuffer()); in Compute() local
1356 U* weights_data = weights_data_tf; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| H A D | operation_selector.cc | 274 std::vector<uint8_t> weights_data(flt_count * SizeOf(weights_desc.type)); in AddConvSharedWeights() local
296 std::vector<uint8_t> weights_data = in AddConvSharedWeights() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/sparse_ops/ |
| H A D | fully_connected.h | 28 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | RowwisePrune.cpp | 42 auto weights_data = weights.data_ptr<scalar_t>(); in _rowwise_prune_helper() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| H A D | fully_connected_op_builder.cc | 58 const float* weights_data = GetTensorData<float>(weights_); in FillCoreMLWeights() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | embedding_lookup_sparse_test.cc | 59 std::initializer_list<float> weights_data) { in SetInput()
|
| H A D | fully_connected_test.cc | 1081 std::vector<float> weights_data(input_depth * output_depth); in SimpleTestQuantizedInt16OutputCase() local
1544 const std::vector<T>& weights_data, in SparseFullyConnectedOpModel()
|
| /aosp_15_r20/external/mesa3d/src/gallium/targets/teflon/ |
| H A D | test_executor.cpp | 129 auto weights_data = &model->buffers[weights_buffer_index]->data; in patch_conv2d() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| H A D | conv_weights_converter_test_util.cc | 67 std::vector<uint8_t> weights_data(flt_count * SizeOf(weights_type)); in ConvolutionWeightsConverterTest() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | lstm_cell.h | 34 const float* weights_data, const RuntimeShape& unextended_bias_shape, in LstmCell()
|