Searched refs:weight_shape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lsh_projection_test.cc | 34 std::initializer_list<int> weight_shape) { in LSHProjectionOpModel() argument
37 if (weight_shape.size() > 0) { in LSHProjectionOpModel()
45 if (weight_shape.size() > 0) { in LSHProjectionOpModel()
46 BuildInterpreter({hash_shape, input_shape, weight_shape}); in LSHProjectionOpModel()
|
| D | embedding_lookup_test.cc | 38 std::initializer_list<int> weight_shape, in BaseEmbeddingLookupOpModel() argument
44 BuildInterpreter({index_shape, weight_shape}); in BaseEmbeddingLookupOpModel()
81 std::initializer_list<int> weight_shape, in HybridEmbeddingLookupOpModel() argument
83 : BaseEmbeddingLookupOpModel(index_shape, weight_shape, type) {} in HybridEmbeddingLookupOpModel()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_weights_test.cc | 40 const TensorShape& weight_shape, in BuildGraphDef() argument
50 Tensor weights_data(DT_FLOAT, weight_shape); in BuildGraphDef()
|
| D | fold_old_batch_norms_test.cc | 372 auto weight_shape = in TestFoldFusedBatchNormsWithConcat() local
374 Tensor weights0_data(DT_FLOAT, weight_shape); in TestFoldFusedBatchNormsWithConcat()
383 Tensor weights1_data(DT_FLOAT, weight_shape); in TestFoldFusedBatchNormsWithConcat()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_batchnorm_test.py | 642 weight_shape = [1] * len(shape)
643 weight_shape[idx] = shape[idx]
645 self.RunWeightedMomentTest(shape, weight_shape, axes, keep_dims, dtype)
648 weight_shape = shape[-(idx + 1):]
650 shape, weight_shape, axes, keep_dims, dtype, dynshapes=dynshapes)
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 1201 std::initializer_list<int> weight_shape) { in LSHProjectionOpModel() argument
1204 if (weight_shape.size() > 0) { in LSHProjectionOpModel()
1212 if (weight_shape.size() > 0) { in LSHProjectionOpModel()
1213 BuildInterpreter({hash_shape, input_shape, weight_shape}); in LSHProjectionOpModel()
3462 std::initializer_list<int> weight_shape, in BaseEmbeddingLookupOpModel() argument
3468 BuildInterpreter({index_shape, weight_shape}); in BaseEmbeddingLookupOpModel()
|