| /external/pytorch/test/edge/ |
| D | test_operator_registration.cpp | 19 EValue* kernel_values[4]; in TEST() local
21 kernel_values[i] = &values[i]; in TEST()
24 op(context, kernel_values); in TEST()
27 ASSERT_TRUE(expected.equal(kernel_values[3]->toTensor())); in TEST()
41 EValue* kernel_values[4]; in TEST() local
43 kernel_values[i] = &values[i]; in TEST()
46 op(context, kernel_values); in TEST()
49 ASSERT_TRUE(expected.equal(kernel_values[3]->toTensor())); in TEST()
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | lstm.py | 70 kernel_values = create_tensor_data(parameters["dtype"],
75 sess.run(tf.group(kernel.assign(kernel_values), bias.assign(bias_values)))
|
| D | static_rnn_with_control_flow_v2.py | 77 kernel_values = create_tensor_data(parameters["dtype"],
82 sess.run(tf.group(kernel.assign(kernel_values), bias.assign(bias_values)))
|
| /external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/ |
| D | static_rnn_with_control_flow_v2.py | 81 kernel_values = create_tensor_data(parameters["dtype"],
86 sess.run(tf.group(kernel.assign(kernel_values), bias.assign(bias_values)))
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| D | fully-connected-sparse.c | 31 const uint8_t* kernel_values, in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8() argument
109 fully_connected->sparse_matrix.values = kernel_values; in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
|
| D | operator-run.c | 83 const uint8_t* kernel_values; member
112 context->kernel_values, in compute_q8gemm_sparse_dq()
142 const uint8_t* kernel_values; member
206 context->kernel_values, in compute_q8gemm_prepacked_sparse_dq()
220 context->kernel_values, in compute_q8gemm_prepacked_sparse_dq()
234 context->kernel_values, in compute_q8gemm_prepacked_sparse_dq()
1125 .kernel_values = op->sparse_matrix.values, in pytorch_qnnp_run_operator()
1201 .kernel_values = op->sparse_matrix.values, in pytorch_qnnp_run_operator()
|
| /external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| D | onedevice_train_palm_detection_filter_inference_beta.cc | 220 const T* __restrict kernel_values, in DepthwiseConv2dNative() argument
292 kernel_values[k_y * kernel_strides[0] + // y in DepthwiseConv2dNative()
311 const T* __restrict kernel_values, in DepthwiseConv2dNativeVALID() argument
318 input_shape, input_values, kernel_shape, kernel_values, stride_y, in DepthwiseConv2dNativeVALID()
326 const T* __restrict kernel_values, in DepthwiseConv2dNativeSAME() argument
332 input_shape, input_values, kernel_shape, kernel_values, stride_y, in DepthwiseConv2dNativeSAME()
|
| D | onedevice_train_palm_detection_filter_inference_v2.cc | 220 const T* __restrict kernel_values, in DepthwiseConv2dNative() argument
292 kernel_values[k_y * kernel_strides[0] + // y in DepthwiseConv2dNative()
311 const T* __restrict kernel_values, in DepthwiseConv2dNativeVALID() argument
318 input_shape, input_values, kernel_shape, kernel_values, stride_y, in DepthwiseConv2dNativeVALID()
326 const T* __restrict kernel_values, in DepthwiseConv2dNativeSAME() argument
332 input_shape, input_values, kernel_shape, kernel_values, stride_y, in DepthwiseConv2dNativeSAME()
|
| D | onedevice_train_palm_detection_filter_inference.cc | 221 const T* __restrict kernel_values, in DepthwiseConv2dNative() argument
293 kernel_values[k_y * kernel_strides[0] + // y in DepthwiseConv2dNative()
312 const T* __restrict kernel_values, in DepthwiseConv2dNativeVALID() argument
319 input_shape, input_values, kernel_shape, kernel_values, stride_y, in DepthwiseConv2dNativeVALID()
327 const T* __restrict kernel_values, in DepthwiseConv2dNativeSAME() argument
333 input_shape, input_values, kernel_shape, kernel_values, stride_y, in DepthwiseConv2dNativeSAME()
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/include/ |
| D | pytorch_qnnpack.h | 180 const uint8_t* kernel_values,
|