Searched defs:weight_dims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | quant_lstm_sup.cc | 30 const TfLiteIntArray* weight_dims, in ExtractQuantLstmWeightsSubmatrix()
47 inline int OutputDepth(const TfLiteIntArray* weight_dims) { in OutputDepth()
51 inline int InputDepth(const TfLiteIntArray* weight_dims) { in InputDepth()
55 void SetWeightSubmatrixDims(const TfLiteIntArray* weight_dims, in SetWeightSubmatrixDims()
82 const TfLiteIntArray* weight_dims, in DecomposeQuantLstmWeightsTensor()
|
| D | quant_lstm_sup_test.cc | 72 const TfLiteIntArray* weight_dims = CreateDimArray(2, {8, 5}); in TEST_F() local
93 const TfLiteIntArray* weight_dims = CreateDimArray(2, {8, 5}); in TEST_F() local
114 const TfLiteIntArray* weight_dims = CreateDimArray(2, {8, 5}); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op_fused.cc | 109 memory::dims weight_dims = memory::dims({channel, k}); in Compute() local
|
| D | mkl_qmatmul_op.cc | 177 memory::dims src_dims, weight_dims; in Compute() local
|
| D | mkl_matmul_ops_common.h | 48 memory::dims weight_dims; member
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 4328 nvinfer1::Dims weight_dims; in TfTensorToTrtWeights() local
|