Home
last modified time | relevance | path

Searched refs:weights_array (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_pure_conv_to_depthwise.cc64 auto& weights_array = model->GetArray(weights_name); in Run() local
65 if (!weights_array.buffer) { in Run()
69 if (weights_array.data_type != ArrayDataType::kFloat) { in Run()
93 depthwiseconv_op->depth_multiplier = weights_array.shape().dims(0); in Run()
98 const auto& weights_shape = weights_array.shape(); in Run()
100 weights_array.GetMutableBuffer<ArrayDataType::kFloat>(); in Run()
113 *weights_array.mutable_shape()->mutable_dims() = {1, width, height, depth}; in Run()
Dshuffle_fc_weights.cc41 Array& weights_array = model->GetArray(weights_name); in Run() local
47 weights_array.data_type != ArrayDataType::kUint8 || in Run()
49 !input_array.quantization_params || !weights_array.quantization_params || in Run()
54 if (!input_array.has_shape() || !weights_array.has_shape()) { in Run()
91 const Shape& weights_shape = weights_array.shape(); in Run()
105 if (!weights_array.buffer) { in Run()
118 weights_array.GetMutableBuffer<ArrayDataType::kUint8>().data; in Run()
Dcreate_im2col_arrays.cc33 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessConvOperator() local
34 if (!weights_array.has_shape()) { in ProcessConvOperator()
40 const auto& weights_shape = weights_array.shape(); in ProcessConvOperator()
Dpropagate_fixed_sizes.cc136 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessConvOperator() local
138 if (!weights_array.has_shape()) { in ProcessConvOperator()
141 const auto& weights_shape = weights_array.shape(); in ProcessConvOperator()
197 const auto& weights_array = in ProcessTransposeConvOperator() local
199 if (!weights_array.has_shape()) { in ProcessTransposeConvOperator()
203 const auto& weights_shape = weights_array.shape(); in ProcessTransposeConvOperator()
263 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessDepthwiseConvOperator() local
265 if (!weights_array.has_shape()) { in ProcessDepthwiseConvOperator()
268 const auto& weights_shape = weights_array.shape(); in ProcessDepthwiseConvOperator()
378 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessFullyConnectedOperator() local
[all …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
Dfuse_binary_into_preceding_affine_test.cc108 auto& weights_array = model_->GetArray(op->inputs[1]); in TEST_F() local
109 EXPECT_THAT(weights_array.GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
Dfuse_binary_into_following_affine_test.cc104 auto& weights_array = model_->GetArray(op->inputs[1]); in TEST_F() local
105 EXPECT_THAT(weights_array.GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.cc690 Operation* weights_array; in ConvertKerasLSTMLayer() local
693 &weights_array))) in ConvertKerasLSTMLayer()
724 /*input_to_input_weights=*/weights_array->getResult(0), in ConvertKerasLSTMLayer()
725 /*input_to_forget_weights=*/weights_array->getResult(1), in ConvertKerasLSTMLayer()
726 /*input_to_cell_weights=*/weights_array->getResult(2), in ConvertKerasLSTMLayer()
727 /*input_to_output_weights=*/weights_array->getResult(3), in ConvertKerasLSTMLayer()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc1904 const auto& weights_array = model.GetArray(op.inputs[1]); in EstimateArithmeticOpsCount() local
1905 if (!output_array.has_shape() || !weights_array.has_shape()) { in EstimateArithmeticOpsCount()
1913 2 * RequiredBufferSizeForShape(weights_array.shape()); in EstimateArithmeticOpsCount()
1923 const auto& weights_array = model.GetArray(op.inputs[1]); in EstimateArithmeticOpsCount() local
1924 if (!input_array.has_shape() || !weights_array.has_shape()) { in EstimateArithmeticOpsCount()
1928 const Shape& weights = weights_array.shape(); in EstimateArithmeticOpsCount()
2427 auto& weights_array = model->GetArray(weights_name); in UndoWeightsShuffling() local
2428 QCHECK(weights_array.data_type == ArrayDataType::kUint8); in UndoWeightsShuffling()
2430 weights_array.GetMutableBuffer<toco::ArrayDataType::kUint8>().data; in UndoWeightsShuffling()
2431 const auto& weights_shape = weights_array.shape(); in UndoWeightsShuffling()
Dexport_tensorflow.cc404 const auto& weights_array = model.GetArray(weights_array_name); in ConvertConvOperator() local
405 CHECK(weights_array.buffer->type == ArrayDataType::kFloat); in ConvertConvOperator()
558 const auto& weights_array = model.GetArray(weights_array_name); in ConvertTransposeConvOperator() local
559 CHECK(weights_array.buffer->type == ArrayDataType::kFloat); in ConvertTransposeConvOperator()
1512 const auto& weights_array = model.GetArray(weights_name); in ConvertLstmCellOperator() local
1514 const auto& weights_shape = weights_array.shape(); in ConvertLstmCellOperator()
1516 CHECK(weights_array.buffer); in ConvertLstmCellOperator()
1517 CHECK(weights_array.buffer->type == ArrayDataType::kFloat); in ConvertLstmCellOperator()
1519 weights_array.GetBuffer<ArrayDataType::kFloat>().data.data(); in ConvertLstmCellOperator()
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dcategory_encoding_test.py107 weights_array = np.array([[.1, .2, .3, .4], [.2, .1, .4, .3]])
109 sparse_weight_data = sparse_ops.from_dense(weights_array)