Searched defs:alpha_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | prelu.h | 30 const T* input_data, const RuntimeShape& alpha_shape, const T* alpha_data, in BroadcastPrelu4DSlow()
77 const T* input_data, const RuntimeShape& alpha_shape, in Prelu()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | prelu_test.cc | 116 const int alpha_shape[] = {3, 1, 1, 3}; in TF_LITE_MICRO_TEST() local
140 const int alpha_shape[] = {3, 1, 1, 3}; in TF_LITE_MICRO_TEST() local
173 const int alpha_shape[] = {3, 1, 1, 3}; in TF_LITE_MICRO_TEST() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | training_ops.cc | 39 TensorShape alpha_shape = ctx->InputShape(1); in Compile() local
84 TensorShape alpha_shape = ctx->InputShape(1); in Compile() local
1044 TensorShape alpha_shape = ctx->InputShape(3); in CheckScalarParams() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 8189 const float* input_data, const RuntimeShape& alpha_shape, in BroadcastPReluDispatch()
|