Home
last modified time | relevance | path

Searched defs:weight_data (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected_test.cc1245 std::initializer_list<float> weight_data = { in TEST_P() local
1272 std::initializer_list<float> weight_data = { in TEST_P() local
1297 std::initializer_list<float> weight_data = { in TEST_P() local
1327 std::initializer_list<float> weight_data = { in TEST_P() local
1360 std::initializer_list<float> weight_data = { in TEST_P() local
1403 std::initializer_list<float> weight_data = { in TEST_P() local
1459 std::initializer_list<float> weight_data = { in TEST_P() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_dimension_numbers_test.cc95 auto weight_data = in XLA_TEST_F() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_matmul_op_fused.cc204 T* weight_data = const_cast<T*>(weight_tensor.flat<T>().data()); in Compute() local
Dmkl_matmul_ops_common.h101 void Execute(const Tinput* src_data, const Tweight* weight_data, in Execute()
427 Tweight* weight_data, const Tensor& weight_tensor, in CacheWeight()
Dmkl_qmatmul_op.cc259 Tweight* weight_data = nullptr; in Compute() local
/external/tensorflow/tensorflow/lite/micro/
Dtest_helpers.cc763 const uint8_t* weight_data = weight->data.uint8; in Invoke() local