Searched defs:weight_data (Results 1 – 6 of 6) sorted by relevance
1646 std::initializer_list<float> weight_data = { in TEST_P() local1673 std::initializer_list<float> weight_data = { in TEST_P() local1701 std::initializer_list<float> weight_data = { in TEST_P() local1726 std::initializer_list<float> weight_data = { in TEST_P() local1756 std::initializer_list<float> weight_data = { in TEST_P() local1786 std::initializer_list<float> weight_data = { in TEST_P() local1820 std::initializer_list<float> weight_data = { in TEST_P() local1864 std::initializer_list<float> weight_data = { in TEST_P() local1926 std::initializer_list<float> weight_data = { in TEST_P() local2046 std::vector<float> weight_data = { in TEST_P() local[all …]
92 auto weight_data = in XLA_TEST_F() local
218 T* weight_data = const_cast<T*>(weight_tensor.flat<T>().data()); in Compute() local
126 void Execute(const Tinput* src_data, const Tweight* weight_data, in Execute()503 Tweight* weight_data, const Tensor& weight_tensor, in CacheWeight()
261 Tweight* weight_data = nullptr; in Compute() local
108 const T* weight_data = weights.data(); in Compute() local