Home
last modified time | relevance | path

Searched defs:output_batch (Results 1 – 20 of 20) sorted by relevance

/aosp_15_r20/external/webrtc/logging/rtc_event_log/events/
H A Drtc_event_definition.h65 rtc::ArrayView<LoggedType> output_batch) const { in ParseImpl()
127 rtc::ArrayView<LoggedType> output_batch = in ParseBatch() local
H A Drtc_event_end_log.cc41 rtc::ArrayView<LoggedStopEvent> output_batch = in Parse() local
H A Drtc_event_begin_log.cc48 rtc::ArrayView<LoggedStartEvent> output_batch = in Parse() local
H A Drtc_event_field_encoding_parser.h284 rtc::ArrayView<E> output_batch = output; in ExtendLoggedBatch() local
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/data/transforms/
H A Dbase.h43 std::vector<Output> output_batch; in apply_batch() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dreduce.h287 const int output_batch = output_shape.Dims(0); in Mean() local
331 const int output_batch = output_shape.Dims(0); in Mean() local
H A Dspace_to_depth.h46 const int output_batch = output_shape.Dims(0); in SpaceToDepth() local
H A Ddepth_to_space.h44 const int output_batch = output_shape.Dims(0); in DepthToSpace() local
H A Dpad.h69 const int output_batch = ext_output_shape.Dims(0); in PadImpl() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A Dmean.h39 const int output_batch = output_shape.Dims(0); in Mean() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dconvolution_dimension_numbers_test.cc36 int64_t input_second_spatial, int64_t output_batch, int64_t output_feature, in CreateConvDimensionNumbers()
H A Dconv_depthwise_backprop_filter_test.cc36 int64_t output_batch, window, window_dilation; member
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dcholesky_op_gpu.cu.cc216 typename TTypes<Scalar, 3>::Tensor output_batch( in ComputeAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A Dmean.h37 const int output_batch = output_shape.Dims(0); in MeanImpl() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/
H A Ddepthwise_planar.hpp367 auto output_batch = reinterpret_cast<TOutput *>(output); in execute_internal() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
H A Ddepthwise_conv_plus_1x1_conv.cc189 int output_batch, int output_width, in Init()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dreduce.h45 const int output_batch = output_shape.Dims(0); in MeanImpl() local
H A Doptimized_ops.h4332 const int output_batch = ext_output_shape.Dims(0); in PadImpl() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlower_tf.cc1022 auto output_batch = padded_shape_splits[0]; in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_a_m.cc397 int64_t output_batch = output_shape[0]; in verify() local