Home
last modified time | relevance | path

Searched defs:output_layout (Results 1 – 25 of 48) sorted by relevance

12

/aosp_15_r20/external/webrtc/audio/utility/
H A Dchannel_mixing_matrix_unittest.cc31 for (ChannelLayout output_layout = CHANNEL_LAYOUT_MONO; in TEST() local
88 ChannelLayout output_layout = CHANNEL_LAYOUT_MONO; in TEST() local
108 ChannelLayout output_layout = CHANNEL_LAYOUT_STEREO; in TEST() local
132 ChannelLayout output_layout = CHANNEL_LAYOUT_2_1; in TEST() local
155 ChannelLayout output_layout = CHANNEL_LAYOUT_2_1; in TEST() local
180 ChannelLayout output_layout = CHANNEL_LAYOUT_5_1; in TEST() local
210 ChannelLayout output_layout = CHANNEL_LAYOUT_5_1; in TEST() local
242 ChannelLayout output_layout = CHANNEL_LAYOUT_7_1; in TEST() local
274 ChannelLayout output_layout = CHANNEL_LAYOUT_7_1; in TEST() local
306 ChannelLayout output_layout = CHANNEL_LAYOUT_MONO; in TEST() local
[all …]
H A Dchannel_mixer_unittest.cc150 for (ChannelLayout output_layout = CHANNEL_LAYOUT_MONO; in TEST_F() local
188 for (ChannelLayout output_layout = CHANNEL_LAYOUT_MONO; in TEST_F() local
H A Dchannel_mixer.cc21 ChannelLayout output_layout) in ChannelMixer()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dconv_spmd_expander.cc290 const Layout& output_layout, ConvBackpropInputOp conv_op) { in HandleConvBackpropInput()
310 const Layout& output_layout, ConvBackpropFilterOp conv_op) { in HandleConvBackpropFilter()
342 const Layout& output_layout, mlir::TF::MaxPoolGradOp max_pool_grad_op) { in HandleMaxPoolGradOp()
484 const Layout output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
H A Dslice_spmd_expander.cc570 const Layout& output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
693 const Layout& output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
837 const Layout& output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
979 const Layout& output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
H A Dmeta_spmd_expander.cc334 const Layout output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
539 const Layout output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
937 const Layout output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
961 const absl::optional<Layout>& output_layout, in RelayoutOneHotInput()
1128 const Layout output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
H A Dgather_spmd_expander.cc249 const Layout output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
338 Status GatherNdGetInputLayoutFromOutput(const Layout& output_layout, in GatherNdGetInputLayoutFromOutput()
529 const Layout output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
H A Dmatmul_spmd_expander.cc190 const Layout& output_layout, std::string& reduced_dim, in MaybeRelayoutInputs()
402 const Layout output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
H A Deinsum_spmd_expander.cc307 const Layout output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
389 const Layout& output_layout, absl::flat_hash_set<std::string>& reduce_dims, in MaybeRelayoutInputs()
H A Dsparse_to_dense_spmd_expander.cc56 Layout output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
H A Didentity_n_spmd_expander.cc38 auto output_layout = layouts[i]; in ExpandOp() local
H A Dqr_spmd_expander.cc99 Layout output_layout = input_layouts.lookup(0).GetLayoutWithReducedDims( in ComputeLayoutForward() local
H A Dcumsum_spmd_expander.cc121 auto output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
H A Dexpanddims_spmd_expander.cc106 auto output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
H A Dsqueeze_spmd_expander.cc84 const Layout& output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
H A Dargmax_spmd_expander.cc152 const Layout& output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
H A Delementwise_spmd_expander.cc149 Layout output_layout = output_layouts.lookup(0); in ComputeLayoutBackward() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ir/
H A Dtf_dtensor.cc90 const tensorflow::dtensor::Layout output_layout = op.output_layout(); in verify() local
146 const tensorflow::dtensor::Layout output_layout = op.output_layout(); in verify() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dstream_executor_util.cc132 std::vector<int64_t> output_layout; in StreamExecutorConvLayoutsToXlaLayouts() local
247 DataLayout output_layout; in XlaConvShapesToStreamExecutorLayouts() local
H A Dmatmul_utils.cc495 MatrixLayout output_layout = config.output_layout; in RunGemm() local
647 MatrixLayout output_layout = config.output_layout; in From() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dvalue_inference_test.cc60 Layout* output_layout = nullptr) { in ComputeDynamismLiteral()
566 XlaOp operand, XlaBuilder* builder, Layout* output_layout = nullptr) { in ComputeUpperBoundLiteral()
729 XlaOp operand, XlaBuilder* builder, Layout* output_layout = nullptr) { in ComputeConstantValueLiteral()
H A Dconv_depthwise_common.h42 std::vector<int64_t> output_layout; member
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dcollectives_common.cc51 const dtensor::Layout& output_layout, in GetAllReducePartitionsFromReducedDims()
/aosp_15_r20/external/webrtc/modules/audio_mixer/
H A Daudio_frame_manipulator.cc80 const ChannelLayout output_layout( in RemixFrame() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/utils/
H A Dcollective_lowering.cc207 StatusOr<Layout> output_layout = in LowerAllReduceOpImpl() local
328 StatusOr<Layout> output_layout = in LowerReduceScatterOp() local

12