| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| 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
|
| 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
|
| 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
|
| D | round_weights_test.cc | 49 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 10})); in TestRoundWeights() local
|
| D | flatten_atrous_test.cc | 64 Tensor weights_data(DT_FLOAT, TensorShape({2, 2, 2, 1})); in TestFlattenAtrousConv() local
|
| D | quantize_weights_test.cc | 50 Tensor weights_data(DT_FLOAT, weight_shape); in BuildGraphDef() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/ |
| 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()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| 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
|
| D | shuffle_fc_weights.cc | 118 auto& weights_data = in Run() local
|
| D | fuse_binary_into_preceding_affine.cc | 130 float* const weights_data = weights_buffer.data.data(); in FuseMulOrDivParamsIntoPrecedingAffine() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/sparse_ops/ |
| D | fully_connected.h | 28 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | RowwisePrune.cpp | 42 auto weights_data = weights.data_ptr<scalar_t>(); in _rowwise_prune_helper() local
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | fully_connected_op_builder.cc | 58 const float* weights_data = GetTensorData<float>(weights_); in FillCoreMLWeights() local
|
| /external/mesa3d/src/gallium/targets/teflon/ |
| D | test_executor.cpp | 136 auto weights_data = &model->buffers[weights_buffer_index]->data; in patch_conv2d() local
301 auto weights_data = &model->buffers[weights_buffer_index]->data; in patch_fully_connected() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse_test.cc | 59 std::initializer_list<float> weights_data) { in SetInput()
|
| 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()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| 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
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| 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
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_weights_converter_test_util.cc | 67 std::vector<uint8_t> weights_data(flt_count * SizeOf(weights_type)); in ConvolutionWeightsConverterTest() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | lstm_cell.h | 34 const float* weights_data, const RuntimeShape& unextended_bias_shape, in LstmCell()
|
| D | fully_connected.h | 32 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
|
| D | legacy_reference_ops.h | 410 const float* weights_data, in FullyConnected()
429 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected()
573 const Dims<4>& prev_activ_dims, const float* weights_data, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 646 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
686 const float* weights_data, in FullyConnected()
705 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected()
1158 const uint8* weights_data, uint8 weights_zero_point, in GEMVForLstmCell()
1346 const RuntimeShape& weights_shape, const uint8* weights_data, in GEMVForLstmCellWithSymmetricRange()
2984 const float* weights_data, const RuntimeShape& unextended_bias_shape, in LstmCell()
3099 const Dims<4>& prev_activ_dims, const float* weights_data, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 2436 auto& weights_data = in UndoWeightsShuffling() local
|
| D | export_tensorflow.cc | 1523 const float* weights_data = in ConvertLstmCellOperator() local
|