Home
last modified time | relevance | path

Searched defs:input_layout (Results 1 – 25 of 29) sorted by relevance

12

/aosp_15_r20/external/webrtc/audio/utility/
H A Dchannel_mixing_matrix_unittest.cc28 for (ChannelLayout input_layout = CHANNEL_LAYOUT_MONO; in TEST() local
87 ChannelLayout input_layout = CHANNEL_LAYOUT_STEREO; in TEST() local
107 ChannelLayout input_layout = CHANNEL_LAYOUT_MONO; in TEST() local
131 ChannelLayout input_layout = CHANNEL_LAYOUT_MONO; in TEST() local
154 ChannelLayout input_layout = CHANNEL_LAYOUT_MONO; in TEST() local
179 ChannelLayout input_layout = CHANNEL_LAYOUT_MONO; in TEST() local
209 ChannelLayout input_layout = CHANNEL_LAYOUT_MONO; in TEST() local
241 ChannelLayout input_layout = CHANNEL_LAYOUT_MONO; in TEST() local
273 ChannelLayout input_layout = CHANNEL_LAYOUT_MONO; in TEST() local
305 ChannelLayout input_layout = CHANNEL_LAYOUT_5_1; in TEST() local
[all …]
H A Dchannel_mixer_unittest.cc147 for (ChannelLayout input_layout = CHANNEL_LAYOUT_MONO; in TEST_F() local
185 for (ChannelLayout input_layout = CHANNEL_LAYOUT_MONO; in TEST_F() local
H A Dchannel_mixer.cc20 ChannelMixer::ChannelMixer(ChannelLayout input_layout, in ChannelMixer()
H A Dchannel_mixing_matrix.cc65 ChannelMixingMatrix::ChannelMixingMatrix(ChannelLayout input_layout, in ChannelMixingMatrix()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dslice_spmd_expander.cc551 const Layout& input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
658 const Layout& input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
796 Layout input_layout; in ComputeLayoutForward() local
942 const Layout& input_layout = input_layouts.lookup(4); in ComputeLayoutForward() local
H A Dmeta_spmd_expander.cc294 const Layout input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
485 const Layout input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
636 const Layout& input_layout, const llvm::ArrayRef<int64_t> output_shape, in MakeLayoutForReshape()
914 const Layout input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
960 Status RelayoutOneHotInput(const absl::optional<Layout>& input_layout, in RelayoutOneHotInput()
H A Dargmax_spmd_expander.cc41 const Layout& input_layout) { in ComputeResultLayout()
126 const Layout& input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
H A Dtop_k_spmd_expander.cc31 StatusOr<Layout> GetSuggestedLayout(const Layout& input_layout) { in GetSuggestedLayout()
H A Dcumsum_spmd_expander.cc108 auto input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
H A Dexpanddims_spmd_expander.cc87 auto input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
H A Dsqueeze_spmd_expander.cc54 const Layout& input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
H A Dbroadcast_to_spmd_expander.cc149 Layout input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
H A Dbias_add_spmd_expander.cc122 Layout input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
H A Dreduce_spmd_expander.cc167 const Layout& input_layout) { in ComputeResultLayout()
H A Dconv_spmd_expander.cc41 Status VerifyConvLayout(const Layout& input_layout, const Layout& filter_layout, in VerifyConvLayout()
H A Dgather_spmd_expander.cc206 int64 dim, bool indices = false) { in ComputeLayoutForward()
H A Dsoftmax_spmd_expander.cc41 const Layout& input_layout, const absl::flat_hash_set<int>& reduced_dims, in ComputeGlobalReduce()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dconv_weights_converter.cc30 Layout input_layout) in ConverterToConvWeights()
242 Layout input_layout) { in CreateConverterToConvWeights()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ir/
H A Dtf_dtensor.cc89 const tensorflow::dtensor::Layout input_layout = op.input_layout(); in verify() local
145 const tensorflow::dtensor::Layout input_layout = op.input_layout(); in verify() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dstream_executor_util.cc70 std::vector<int64_t> input_layout; in StreamExecutorConvLayoutsToXlaLayouts() local
198 DataLayout input_layout; in XlaConvShapesToStreamExecutorLayouts() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtranspose.cc883 std::variant<Tiling, Striding> input_layout, Tiling output_tiling, in Create()
1333 absl::InlinedVector<int64_t, 4> input_layout; member
1368 std::variant<TransposePlan::Tiling, TransposePlan::Striding> input_layout, in GetOrCreate()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
H A Dconvolution_selector.cc129 ModelHints hints, Layout input_layout) { in SelectConverterToConvWeights()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dlayout_propagation_v2.cc618 const auto input_layout = new_input_layouts.find(i); in UpdateLayoutsForOp() local
1157 const Layout input_layout = in InsertRelayoutForWhileLoops() local
/aosp_15_r20/external/mesa3d/src/gallium/drivers/d3d12/
H A Dd3d12_pipeline_state.cpp346 D3D12_INPUT_LAYOUT_DESC& input_layout = (D3D12_INPUT_LAYOUT_DESC&)pso_desc.InputLayout; in create_gfx_pipeline_state() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
H A Doperation_selector.cc234 Layout input_layout = Layout::OHWI; in AddDynamicConv() local

12