Home
last modified time | relevance | path

Searched refs:output_layout (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc94 std::vector<int64> output_layout; in AddBackendConstraintsToDnnConvCustomCall() local
97 output_layout.push_back(dimension_numbers.output_spatial_dimensions(i)); in AddBackendConstraintsToDnnConvCustomCall()
99 output_layout.push_back(dimension_numbers.output_feature_dimension()); in AddBackendConstraintsToDnnConvCustomCall()
100 output_layout.push_back(dimension_numbers.output_batch_dimension()); in AddBackendConstraintsToDnnConvCustomCall()
101 *output_shape.mutable_layout() = LayoutUtil::MakeLayout(output_layout); in AddBackendConstraintsToDnnConvCustomCall()
/external/tensorflow/tensorflow/core/kernels/
Dmkl_tfconv_op.h166 dnnLayout_t output_layout = in ConvertMklToTf() local
174 input_shape.GetConvertedFlatData(output_layout, input_buffer, in ConvertMklToTf()
Dmkl_reshape_op.cc118 dnnLayout_t output_layout = in Compute() local
127 mkl_shape_input.GetConvertedFlatData(output_layout, input_buffer, in Compute()
/external/webrtc/webrtc/modules/audio_processing/include/
Dmock_audio_processing.h187 ChannelLayout output_layout,
221 ChannelLayout output_layout,
Daudio_processing.h281 ChannelLayout output_layout,
333 ChannelLayout output_layout,
/external/webrtc/webrtc/modules/audio_processing/
Daudio_processing_impl.h55 ChannelLayout output_layout,
73 ChannelLayout output_layout,
Daudio_processing_impl.cc297 ChannelLayout output_layout, in Initialize() argument
304 ChannelsFromLayout(output_layout), in Initialize()
305 LayoutHasKeyboard(output_layout)}, in Initialize()
562 ChannelLayout output_layout, in ProcessStream() argument
580 output_stream.set_num_channels(ChannelsFromLayout(output_layout)); in ProcessStream()
581 output_stream.set_has_keyboard(LayoutHasKeyboard(output_layout)); in ProcessStream()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc1328 std::vector<int64> output_layout; in AlignLayouts() local
1329 output_layout.reserve(output_rank); in AlignLayouts()
1384 output_layout.push_back(j); in AlignLayouts()
1392 output_layout.push_back(i); in AlignLayouts()
1395 CHECK_EQ(output_layout.size(), output_rank); in AlignLayouts()
1398 output_layout); in AlignLayouts()
Dshape_util_test.cc627 Layout output_layout = output_is_row_major in TEST() local
650 F32, {6, 2}, AsInt64Slice(output_layout.minor_to_major()))), in TEST()
Dxla.proto395 Layout output_layout = 3; field
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcompute_constant_test.cc74 ComputationBuilder* builder, Layout* output_layout = nullptr, in ComputeConstantLiteral() argument
77 operand, output_layout, parameters));
Dreduce_window_test.cc328 std::vector<int64> output_layout = {1, 5, 3, 2, 0, 4}; in XLA_TEST_P() local
331 ShapeUtil::MakeShapeWithLayout(F32, output_dims, output_layout); in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc891 const Layout& output_layout, const HloInstruction* instruction, in ChooseOperandLayoutFromOutputLayout() argument
908 return MakeUnique<Layout>(output_layout); in ChooseOperandLayoutFromOutputLayout()
926 LayoutUtil::MinorToMajor(output_layout)); in ChooseOperandLayoutFromOutputLayout()
934 *operand_shape.mutable_layout() = output_layout; in ChooseOperandLayoutFromOutputLayout()
937 return MakeUnique<Layout>(output_layout); in ChooseOperandLayoutFromOutputLayout()
955 int64 output_dim = LayoutUtil::Minor(output_layout, i); in ChooseOperandLayoutFromOutputLayout()
Dlayout_assignment.h344 const Layout& output_layout, const HloInstruction* instruction,
Dservice.cc1251 arg->output_layout(), execution_options.shape_with_output_layout())); in ComputeConstant()
1253 arg->output_layout(); in ComputeConstant()
1280 result_literal = result_literal->Relayout(arg->output_layout()); in ComputeConstant()
/external/webrtc/webrtc/modules/audio_processing/test/
Dprocess_test.cc582 AudioProcessing::ChannelLayout output_layout = AudioProcessing::kMono; in void_main() local
605 output_layout = in void_main()
614 output_layout, in void_main()
762 output_layout, in void_main()
/external/webrtc/talk/media/webrtc/
Dfakewebrtcvoiceengine.h68 webrtc::AudioProcessing::ChannelLayout output_layout,
92 webrtc::AudioProcessing::ChannelLayout output_layout,
/external/tensorflow/tensorflow/compiler/xla/client/
Dcomputation_builder.cc1160 const ComputationDataHandle& operand, const Layout* output_layout, in ComputeConstant() argument
1167 if (output_layout != nullptr) { in ComputeConstant()
1168 *request.mutable_output_layout() = *output_layout; in ComputeConstant()
Dcomputation_builder.h818 const Layout* output_layout = nullptr,
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h692 dnnLayout_t output_layout = static_cast<dnnLayout_t>(mkl_shape.GetTfLayout()); in ConvertMklToTF() local
697 mkl_shape.GetConvertedFlatData(output_layout, input_buffer, output_buffer); in ConvertMklToTF()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2346 const Layout& output_layout = output_shape.layout(); in EmitFastConcatenate() local
2347 auto output_min2maj = LayoutUtil::MinorToMajor(output_layout); in EmitFastConcatenate()