Home
last modified time | relevance | path

Searched defs:weight_shape (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dembedding_lookup_test.cc42 std::initializer_list<int> weight_shape, in BaseEmbeddingLookupOpModel()
91 std::initializer_list<int> weight_shape, in HybridEmbeddingLookupOpModel()
Dlsh_projection_test.cc35 std::initializer_list<int> weight_shape) { in LSHProjectionOpModel()
Dquant_basic_lstm_test.cc47 std::vector<int> weight_shape{4 * outputSize, outputSize + inputSize}; in QuantizedLSTMOpModel() local
Dunidirectional_sequence_lstm.cc686 const RuntimeShape& weight_shape = GetTensorShape(weight_tensor); in PrecomputeZeroPointTimesWeightWithBias() local
Dlstm.cc1133 const RuntimeShape& weight_shape = GetTensorShape(weight_tensor); in PrecomputeZeroPointTimesWeightWithBias() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_weights_test.cc40 const TensorShape& weight_shape, in BuildGraphDef()
Dfold_old_batch_norms_test.cc373 auto weight_shape = in TestFoldFusedBatchNormsWithConcat() local
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils_test.cc46 SmallVector<int64_t, 2> weight_shape{3, 12}; in createLstmCompositeFunc() local
/external/tensorflow/tensorflow/core/grappler/utils/
Dpattern_utils_test.cc35 auto weight_shape = ops::Placeholder::Shape({32, 64}); in GetMatMulBiasAddGeluGraph() local
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2024 std::initializer_list<int> weight_shape) { in LSHProjectionOpModel()
4753 std::initializer_list<int> weight_shape, in BaseEmbeddingLookupOpModel()