| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | fully_connected.cc | 40 FullyConnected::FullyConnected(const OperationDef& definition, in FullyConnected() function in tflite::gpu::FullyConnected
60 FullyConnected::FullyConnected(FullyConnected&& kernel) in FullyConnected() function in tflite::gpu::FullyConnected
63 FullyConnected& FullyConnected::operator=(FullyConnected&& kernel) { in operator =()
76 std::string FullyConnected::GetFullyConnectedKernelCode( in GetFullyConnectedKernelCode()
153 int3 FullyConnected::GetGridSize() const { in GetGridSize()
157 FullyConnected CreateFullyConnected(const GpuInfo& gpu_info, in CreateFullyConnected()
160 FullyConnected result(definition, gpu_info); in CreateFullyConnected()
|
| D | fully_connected.h | 116 class FullyConnected : public GPUOperation {
118 FullyConnected() = default;
128 FullyConnected(FullyConnected&& kernel);
129 FullyConnected& operator=(FullyConnected&& kernel);
130 FullyConnected(const FullyConnected&) = delete;
131 FullyConnected& operator=(const FullyConnected&) = delete;
134 FullyConnected(const OperationDef& definition, const GpuInfo& gpu_info);
135 friend FullyConnected CreateFullyConnected(
148 void FullyConnected::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeights()
194 FullyConnected CreateFullyConnected(const GpuInfo& gpu_info,
|
| D | fully_connected_test_util.cc | 49 FullyConnected operation = in FullyConnectedTest()
52 src_tensor, absl::make_unique<FullyConnected>(std::move(operation)), in FullyConnectedTest()
95 FullyConnected operation = in FullyConnectedLargeTest()
98 src_tensor, absl::make_unique<FullyConnected>(std::move(operation)), in FullyConnectedLargeTest()
152 FullyConnected operation = in FullyConnectedExtraLargeTest()
155 src_tensor, absl::make_unique<FullyConnected>(std::move(operation)), in FullyConnectedExtraLargeTest()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/ |
| D | fully_connected_selector.cc | 35 FullyConnected fc = CreateFullyConnected(gpu_info, op_def, attr); in SelectFullyConnectedGeneric()
36 return absl::make_unique<FullyConnected>(std::move(fc)); in SelectFullyConnectedGeneric()
48 FullyConnected fc = CreateFullyConnected(gpu_info, op_def, attr); in SelectFullyConnectedAdreno()
49 return absl::make_unique<FullyConnected>(std::move(fc)); in SelectFullyConnectedAdreno()
60 FullyConnected fc = CreateFullyConnected(gpu_info, op_def, attr); in SelectFullyConnectedPowerVR()
61 return absl::make_unique<FullyConnected>(std::move(fc)); in SelectFullyConnectedPowerVR()
79 FullyConnected fc = CreateFullyConnected(gpu_info, op_def, attr); in SelectFullyConnectedMali()
80 return absl::make_unique<FullyConnected>(std::move(fc)); in SelectFullyConnectedMali()
101 FullyConnected fc = CreateFullyConnected(gpu_info, op_def, attr); in SelectFullyConnected()
102 return absl::make_unique<FullyConnected>(std::move(fc)); in SelectFullyConnected()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | fully_connected_test.cc | 28 TEST(FullyConnected, 1D) {
47 TEST(FullyConnected, 1DKeepDims) {
67 TEST(FullyConnected, 2D) {
89 TEST(FullyConnected, 2DKeepDims) {
112 TEST(FullyConnected, 3D) {
135 TEST(FullyConnected, 3DReshape) {
158 TEST(FullyConnected, 3DKeepDims) {
182 TEST(FullyConnected, 4D) {
206 TEST(FullyConnected, 4DKeepDims) {
231 TEST(FullyConnected, FP16Weights) { in TEST() argument
[all …]
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | gru_cell.cc | 28 using optimized_ops::FullyConnected;
63 FullyConnected(fc_params, concat_shape, concat, gate_weight_shape, in GruCell()
80 FullyConnected(fc_params, concat_shape, concat, candidate_weight_shape, in GruCell()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | fully_connected.cc | 85 tflite::reference_ops::FullyConnected( in Eval()
99 tflite::reference_integer_ops::FullyConnected( in Eval()
113 tflite::reference_ops::FullyConnected( in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | fully_connected.h | 23 inline void FullyConnected( in FullyConnected() function
65 inline void FullyConnected( in FullyConnected() function
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa/ |
| D | fully_connected.cc | 34 void FullyConnected(const FullyConnectedParams& params, in FullyConnected() function
196 FullyConnected(FullyConnectedParamsQuantized(data), in EvalQuantizedInt8()
236 reference_integer_ops::FullyConnected( in EvalQuantizedInt8()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fully_connected.cc | 649 reference_integer_ops::FullyConnected( in FullyConnectedInt8()
655 optimized_integer_ops::FullyConnected( in FullyConnectedInt8()
676 reference_integer_ops::FullyConnected( in FullyConnectedInt16()
727 reference_ops::FullyConnected( in EvalQuantized()
733 optimized_ops::FullyConnected( in EvalQuantized()
750 reference_ops::FullyConnected( in EvalQuantized()
756 optimized_ops::FullyConnected( in EvalQuantized()
849 reference_ops::FullyConnected( in EvalFloat()
896 optimized_ops::FullyConnected( in EvalFloat()
|
| /external/tensorflow/tensorflow/python/layers/ |
| D | core.py | 36 FullyConnected = Dense variable
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | fully_connected.h | 26 inline void FullyConnected( in FullyConnected() function
62 inline void FullyConnected( in FullyConnected() function
110 inline void FullyConnected( in FullyConnected() function
|
| D | legacy_reference_ops.h | 407 inline void FullyConnected(const float* input_data, const Dims<4>& input_dims, in FullyConnected() function
418 FullyConnected(op_params, DimsToShape(input_dims), input_data, in FullyConnected()
426 void FullyConnected(const float* input_data, const Dims<4>& input_dims, in FullyConnected() function
432 FullyConnected(input_data, input_dims, weights_data, weights_dims, bias_data, in FullyConnected()
437 inline void FullyConnected( in FullyConnected() function
443 FullyConnected(params, input_shape, input_data, filter_shape, filter_data, in FullyConnected()
447 inline void FullyConnected( in FullyConnected() function
453 FullyConnected(params, input_shape, input_data, filter_shape, filter_data, in FullyConnected()
457 inline void FullyConnected(const uint8* input_data, const Dims<4>& input_dims, in FullyConnected() function
476 FullyConnected(op_params, DimsToShape(input_dims), input_data, in FullyConnected()
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/ |
| D | fully_connected.cc | 170 tflite::reference_integer_ops::FullyConnected( in EvalQuantizedInt8()
204 tflite::reference_ops::FullyConnected( in Eval()
221 tflite::reference_ops::FullyConnected( in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/sparse_ops/ |
| D | fully_connected.h | 39 FullyConnected(params, input_shape, input_data, weights_shape, in FullyConnectedSparseWeight()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | fully_connected_test.cc | 38 TEST_F(OpenCLOperationTest, FullyConnected) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/ |
| D | fully_connected.cc | 273 reference_integer_ops::FullyConnected( in EvalQuantizedInt8()
311 reference_ops::FullyConnected( \ in EvalQuantized()
354 tflite::reference_ops::FullyConnected( in EvalFloat()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | fully_connected.h | 30 inline void FullyConnected( in FullyConnected() function
|
| /external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| D | core.py | 338 FullyConnected = Dense variable
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | README.md | 77 * FullyConnected
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_v3a.fbs | 559 // Parameters for FullyConnected version 1 or above.
562 // Parameters for FullyConnected version 2 or above.
565 // Parameters for FullyConnected version 5 or above.
570 // Parameters for FullyConnected version 7 or above.
|
| D | schema.fbs | 587 // Parameters for FullyConnected version 1 or above.
590 // Parameters for FullyConnected version 2 or above.
593 // Parameters for FullyConnected version 5 or above.
598 // Parameters for FullyConnected version 7 or above.
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | test_schema.fbs | 555 // Parameters for FullyConnected version 1 or above.
558 // Parameters for FullyConnected version 2 or above.
561 // Parameters for FullyConnected version 5 or above.
566 // Parameters for FullyConnected version 7 or above.
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 642 inline void FullyConnected( in FullyConnected() function
684 inline void FullyConnected(const float* input_data, const Dims<4>& input_dims, in FullyConnected() function
695 FullyConnected(op_params, DimsToShape(input_dims), input_data, in FullyConnected()
703 void FullyConnected(const float* input_data, const Dims<4>& input_dims, in FullyConnected() function
709 FullyConnected(input_data, input_dims, weights_data, weights_dims, bias_data, in FullyConnected()
1090 inline void FullyConnected( in FullyConnected() function
1634 inline void FullyConnected( in FullyConnected() function
1721 inline void FullyConnected(const uint8* input_data, const Dims<4>& input_dims, in FullyConnected() function
1740 FullyConnected(op_params, DimsToShape(input_dims), input_data, in FullyConnected()
1746 inline void FullyConnected( in FullyConnected() function
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/ |
| D | README.md | 75 * FullyConnected (aka Dense or InnerProduct)
|