Home
last modified time | relevance | path

Searched refs:weight_dims (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dquant_lstm_sup.cc30 const TfLiteIntArray* weight_dims, in ExtractQuantLstmWeightsSubmatrix() argument
35 auto const& weight_cols = weight_dims->data[1]; in ExtractQuantLstmWeightsSubmatrix()
47 inline int OutputDepth(const TfLiteIntArray* weight_dims) { in OutputDepth() argument
48 return weight_dims->data[0] / 4; in OutputDepth()
51 inline int InputDepth(const TfLiteIntArray* weight_dims) { in InputDepth() argument
52 return weight_dims->data[1] - OutputDepth(weight_dims); in InputDepth()
55 void SetWeightSubmatrixDims(const TfLiteIntArray* weight_dims, in SetWeightSubmatrixDims() argument
58 const auto input_depth = InputDepth(weight_dims); in SetWeightSubmatrixDims()
59 const auto output_depth = OutputDepth(weight_dims); in SetWeightSubmatrixDims()
82 const TfLiteIntArray* weight_dims, in DecomposeQuantLstmWeightsTensor() argument
[all …]
Dquant_lstm_sup.h29 const TfLiteIntArray* weight_dims,
34 const TfLiteIntArray* weight_dims,
44 void SetWeightSubmatrixDims(const TfLiteIntArray* weight_dims,
Dquant_lstm_sup_test.cc72 const TfLiteIntArray* weight_dims = CreateDimArray(2, {8, 5}); in TEST_F() local
78 0 /* offset_column */, weight_dims, in TEST_F()
93 const TfLiteIntArray* weight_dims = CreateDimArray(2, {8, 5}); in TEST_F() local
99 3 /* offset_column */, weight_dims, in TEST_F()
114 const TfLiteIntArray* weight_dims = CreateDimArray(2, {8, 5}); in TEST_F() local
121 3 /* offset_column */, weight_dims, weights.data(), &submatrix); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_qmatmul_op.cc177 memory::dims src_dims, weight_dims; in Compute() local
189 weight_dims = TFShapeToMklDnnDims(weight_tf_shape); in Compute()
195 weight_dims = {static_cast<int>(weight_tf_shape.dim_size(1)), in Compute()
219 : memory::desc(weight_dims, MklDnnType<Tweight>(), in Compute()
227 MklDnnMatMulFwdParams matmul_fwd_dims(src_dims, weight_dims, bias_dims, in Compute()
Dmkl_matmul_op_fused.cc109 memory::dims weight_dims = memory::dims({channel, k}); in Compute() local
120 src_dims, weight_dims, bias_dims, dst_dims, src_format, in Compute()
225 memory::desc(weight_dims, MklDnnType<T>(), weight_format); in Compute()
Dmkl_matmul_ops_common.h48 memory::dims weight_dims; member
61 MklDnnMatMulFwdParams(memory::dims src_dims, memory::dims weight_dims,
67 weight_dims(weight_dims), in src_dims()
183 context_.weight_md.reset(new memory::desc({matmul_fwd_params.weight_dims}, in Setup()
337 key_creator.AddAsKey(mkldnn_matmul_fwd_dims.weight_dims); in CreateKey()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc4328 nvinfer1::Dims weight_dims; in TfTensorToTrtWeights() local
4329 GetTensorDimsWithProtoShape(tensor, &weight_dims); in TfTensorToTrtWeights()
4330 *weights = weight_store->GetTempWeights(trt_dtype, weight_dims); in TfTensorToTrtWeights()