Home
last modified time | relevance | path

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

12

/external/webrtc/audio/utility/
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 …]
Dchannel_mixer.cc20 ChannelMixer::ChannelMixer(ChannelLayout input_layout, in ChannelMixer()
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
Dchannel_mixing_matrix.cc65 ChannelMixingMatrix::ChannelMixingMatrix(ChannelLayout input_layout, in ChannelMixingMatrix()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter.cc30 Layout input_layout) in ConverterToConvWeights()
242 Layout input_layout) { in CreateConverterToConvWeights()
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dargmax_spmd_expander.cc41 const Layout& input_layout) { in ComputeResultLayout()
126 const Layout& input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
Dtop_k_spmd_expander.cc31 StatusOr<Layout> GetSuggestedLayout(const Layout& input_layout) { in GetSuggestedLayout()
Dcumsum_spmd_expander.cc108 auto input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
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
Dexpanddims_spmd_expander.cc87 auto input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
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()
Dsqueeze_spmd_expander.cc54 const Layout& input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
Dbroadcast_to_spmd_expander.cc149 Layout input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
Dbias_add_spmd_expander.cc122 Layout input_layout = input_layouts.lookup(0); in ComputeLayoutForward() local
Dreduce_spmd_expander.cc167 const Layout& input_layout) { in ComputeResultLayout()
Dconv_spmd_expander.cc41 Status VerifyConvLayout(const Layout& input_layout, const Layout& filter_layout, in VerifyConvLayout()
Dgather_spmd_expander.cc206 int64 dim, bool indices = false) { in ComputeLayoutForward()
Dsoftmax_spmd_expander.cc41 const Layout& input_layout, const absl::flat_hash_set<int>& reduced_dims, in ComputeGlobalReduce()
/external/tensorflow/tensorflow/dtensor/mlir/ir/
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
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_executor_util.cc70 std::vector<int64_t> input_layout; in StreamExecutorConvLayoutsToXlaLayouts() local
198 DataLayout input_layout; in XlaConvShapesToStreamExecutorLayouts() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dconvolution_selector.cc129 ModelHints hints, Layout input_layout) { in SelectConverterToConvWeights()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
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()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Doperation_selector.cc234 Layout input_layout = Layout::OHWI; in AddDynamicConv() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dslice_test.cc476 std::array<int64_t, 4> input_layout; // minor-to-major member
/external/tensorflow/tensorflow/dtensor/mlir/
Dlayout_propagation_v2.cc618 const auto input_layout = new_input_layouts.find(i); in UpdateLayoutsForOp() local
1157 const Layout input_layout = in InsertRelayoutForWhileLoops() local

12