| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fully_connected_test.cc | 1245 std::initializer_list<float> weight_data = { in TEST_P() local
1257 /*input=*/{TensorType_FLOAT32, {2, 10}}, weight, weight_data); in TEST_P()
1272 std::initializer_list<float> weight_data = { in TEST_P() local
1282 /*input=*/{TensorType_FLOAT32, {2, 2}}, weight, weight_data); in TEST_P()
1297 std::initializer_list<float> weight_data = { in TEST_P() local
1312 weight, weight_data); in TEST_P()
1327 std::initializer_list<float> weight_data = { in TEST_P() local
1343 /*input=*/{TensorType_FLOAT32, {2, 12}}, weight, weight_data, in TEST_P()
1360 std::initializer_list<float> weight_data = { in TEST_P() local
1376 /*input=*/{TensorType_FLOAT32, {6, 12}}, weight, weight_data, in TEST_P()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op_fused.cc | 204 T* weight_data = const_cast<T*>(weight_tensor.flat<T>().data()); in Compute() local
242 weight_data = cached_weight_data; in Compute()
244 weight_mkl.SetUsrMem(weight_md, weight_data); in Compute()
247 weight_data = in Compute()
254 matmul_prim->Execute(src_data, weight_data, bias_data, dst_data, in Compute()
|
| D | mkl_qmatmul_op.cc | 259 Tweight* weight_data = nullptr; in Compute() local
270 this->CacheWeight(context, matmul_fwd_pd, weight_data, in Compute()
273 weight_data = in Compute()
275 is_weight_cached = (weight_data != nullptr); in Compute()
282 weight_data = in Compute()
287 weight_data = static_cast<Tweight*>( in Compute()
296 matmul_fwd->Execute(src_data, weight_data, bias_data, dst_data, in Compute()
|
| D | mkl_matmul_ops_common.h | 101 void Execute(const Tinput* src_data, const Tweight* weight_data, in Execute() argument
108 static_cast<void*>(const_cast<Tweight*>(weight_data)), *fwd_stream); in Execute()
117 static_cast<void*>(const_cast<Tweight*>(weight_data))); in Execute()
427 Tweight* weight_data, const Tensor& weight_tensor, in CacheWeight() argument
442 weight_data = static_cast<Tweight*>(weight.GetOpMem().get_data_handle()); in CacheWeight()
455 memcpy(weight_oi_t_data, weight_data, weight_size); in CacheWeight()
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | category_encoding_test.py | 119 weight_data = keras.Input(shape=(None,), dtype=dtypes.float32, sparse=True)
123 int_data = layer(input_data, count_weights=weight_data)
126 model = keras.Model(inputs=[input_data, weight_data], outputs=int_data)
175 weight_data = keras.Input(shape=(None,), dtype=dtypes.float32, sparse=True)
188 int_data = layer(input_data, count_weights=weight_data)
190 model = keras.Model(inputs=[input_data, weight_data], outputs=int_data)
|
| D | index_lookup_test.py | 1188 weight_data = [1, 1, 1, 1, 1]
1198 layer.set_vocabulary(vocab_data, idf_weights=weight_data)
1202 weight_data = [1, 1, 1, 1, 1] # too long
1212 layer.set_vocabulary(vocab_data, idf_weights=weight_data)
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_dimension_numbers_test.cc | 95 auto weight_data = in XLA_TEST_F() local
127 ComputeAndCompareR4<float>(&builder, *expected_conv2, {weight_data.get()}, in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_utils_test.py | 61 weight_data = {key: value for key, value in zip(weights[0], weights[1])}
62 self.assertDictEqual(table_data, weight_data)
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | test_helpers.cc | 763 const uint8_t* weight_data = weight->data.uint8; in Invoke() local
769 output_data[0] = input_data[0] + weight_data[0]; in Invoke()
|