Home
last modified time | relevance | path

Searched refs:ExtendedShape (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h51 auto extended_shape0 = RuntimeShape::ExtendedShape(dims_count, shape0); in ProcessBroadcastShapes()
52 auto extended_shape1 = RuntimeShape::ExtendedShape(dims_count, shape1); in ProcessBroadcastShapes()
298 RuntimeShape::ExtendedShape(4, unextended_input_shape); in DepthToSpace()
300 RuntimeShape::ExtendedShape(4, unextended_output_shape); in DepthToSpace()
351 RuntimeShape::ExtendedShape(4, unextended_input_shape); in SpaceToDepth()
353 RuntimeShape::ExtendedShape(4, unextended_output_shape); in SpaceToDepth()
706 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastAdd4DSlow()
747 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastAdd4DSlow()
789 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastAdd4DSlow()
976 RuntimeShape::ExtendedShape(4, unextended_output_shape); in BroadcastMul4DSlow()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmean.h33 RuntimeShape::ExtendedShape(4, unextended_input_shape); in Mean()
35 RuntimeShape::ExtendedShape(4, unextended_output_shape); in Mean()
Dmul.h100 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastMul4DSlow()
Dadd.h90 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastAdd4DSlow()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dcommon.h235 auto extended_input0_shape = RuntimeShape::ExtendedShape(N, input0_shape); in NdArrayDescsForElementwiseBroadcast()
236 auto extended_input1_shape = RuntimeShape::ExtendedShape(N, input1_shape); in NdArrayDescsForElementwiseBroadcast()
Dtypes.h231 inline static RuntimeShape ExtendedShape(int new_shape_size, in ExtendedShape() function
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1930 RuntimeShape::ExtendedShape(4, unextended_input_shape); in Mean()
1932 RuntimeShape::ExtendedShape(4, unextended_output_shape); in Mean()
2480 RuntimeShape::ExtendedShape(4, unextended_input_shape); in DepthToSpace()
2482 RuntimeShape::ExtendedShape(4, unextended_output_shape); in DepthToSpace()
2521 RuntimeShape::ExtendedShape(4, unextended_input_shape); in SpaceToDepth()
2523 RuntimeShape::ExtendedShape(4, unextended_output_shape); in SpaceToDepth()
3444 RuntimeShape::ExtendedShape(4, unextended_output_shape); in BroadcastDiv4DSlow()
3574 RuntimeShape::ExtendedShape(4, unextended_input_shape); in LstmCell()
3576 RuntimeShape::ExtendedShape(4, unextended_prev_activ_shape); in LstmCell()
3578 RuntimeShape::ExtendedShape(4, unextended_bias_shape); in LstmCell()
[all …]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h2648 RuntimeShape::ExtendedShape(4, unextended_input_shape); in DepthToSpace()
2650 RuntimeShape::ExtendedShape(4, unextended_output_shape); in DepthToSpace()
2689 RuntimeShape::ExtendedShape(4, unextended_input_shape); in SpaceToDepth()
2691 RuntimeShape::ExtendedShape(4, unextended_output_shape); in SpaceToDepth()
3520 RuntimeShape::ExtendedShape(4, unextended_output_shape); in BroadcastDiv4DSlow()
3665 RuntimeShape::ExtendedShape(4, unextended_input_shape); in LstmCell()
3667 RuntimeShape::ExtendedShape(4, unextended_prev_activ_shape); in LstmCell()
3669 RuntimeShape::ExtendedShape(4, unextended_bias_shape); in LstmCell()
3671 RuntimeShape::ExtendedShape(4, unextended_prev_state_shape); in LstmCell()
3673 RuntimeShape::ExtendedShape(4, unextended_output_state_shape); in LstmCell()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h516 auto extended_input0_shape = RuntimeShape::ExtendedShape(N, input0_shape); in NdArrayDescsForElementwiseBroadcast()
517 auto extended_input1_shape = RuntimeShape::ExtendedShape(N, input1_shape); in NdArrayDescsForElementwiseBroadcast()
Dtypes.h246 inline static RuntimeShape ExtendedShape(int new_shape_size, in ExtendedShape() function