| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | stream_executor_util.cc | 85 std::vector<int64> output_layout; in StreamExecutorConvLayoutsToXlaLayouts() local
88 output_layout.push_back(dnums.output_batch_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
89 output_layout.push_back(dnums.output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
90 output_layout.insert(output_layout.end(), in StreamExecutorConvLayoutsToXlaLayouts()
95 output_layout.push_back(dnums.output_batch_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
96 output_layout.insert(output_layout.end(), in StreamExecutorConvLayoutsToXlaLayouts()
99 output_layout.push_back(dnums.output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
109 LayoutUtil::MakeLayoutFromMajorToMinor(output_layout)); in StreamExecutorConvLayoutsToXlaLayouts()
152 DataLayout output_layout; in XlaConvLayoutsToStreamExecutorLayouts() local
154 output_layout = DataLayout::kBatchDepthYX; in XlaConvLayoutsToStreamExecutorLayouts()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conv_depthwise_test.cc | 42 std::vector<int64> output_layout; member
92 config.output_layout = {3, 0, 2, 1}; in GetConv2DTestCases()
100 config.output_layout = {0, 3, 2, 1}; in GetConv2DTestCases()
122 absl::StrJoin(spec.output_layout, "_"), data_type); in DepthwiseConvolution2DTestDataToString()
154 absl::StrJoin(spec.output_layout, ","), data_type, in BuildHloTextDepthwiseConvolution2D()
179 absl::StrJoin(spec.output_layout, ","), data_type, in BuildHloTextDepthwiseConvolution2D()
203 absl::StrJoin(spec.output_layout, ","), data_type, in BuildHloTextDepthwiseConvolution2D()
|
| D | grouped_convolution_test.cc | 43 std::vector<int64> output_layout; member
101 config.output_layout = {3, 0, 2, 1}; in GetConv2DTestCases()
109 config.output_layout = {0, 3, 2, 1}; in GetConv2DTestCases()
131 absl::StrJoin(spec.output_layout, "_"), data_type); in GroupedConvolution2DTestDataToString()
164 absl::StrJoin(spec.output_layout, ","), data_type, in BuildHloTextGroupedConvolution2D()
190 absl::StrJoin(spec.output_layout, ","), data_type, in BuildHloTextGroupedConvolution2D()
215 absl::StrJoin(spec.output_layout, ","), data_type, in BuildHloTextGroupedConvolution2D()
|
| D | conv_depthwise_backprop_filter_test.cc | 41 std::vector<int64> output_layout; member
75 config.output_layout = {2, 3, 0, 1}; in GetConv2DTestCases()
92 config.output_layout = {2, 3, 0, 1}; in GetConv2DTestCases()
119 absl::StrJoin(spec.output_layout, "_")); in BatchGroupedConvolution2DTestDataToString()
146 absl::StrJoin(spec.output_layout, ","), data_type, in BuildHloTextBatchGroupedConvolution2D()
|
| D | compute_constant_test.cc | 74 Layout* output_layout = nullptr) { in ComputeConstantLiteral() argument
77 client->ComputeConstant(subgraph, output_layout)); in ComputeConstantLiteral()
|
| D | reduce_window_test.cc | 369 std::vector<int64> output_layout = {1, 5, 3, 2, 0, 4}; in XLA_TEST_P() local
372 ShapeUtil::MakeShapeWithLayout(F32, output_dims, output_layout); in XLA_TEST_P()
|
| /external/webrtc/webrtc/modules/audio_processing/include/ |
| D | mock_audio_processing.h | 187 ChannelLayout output_layout,
221 ChannelLayout output_layout,
|
| D | audio_processing.h | 281 ChannelLayout output_layout,
333 ChannelLayout output_layout,
|
| /external/webrtc/webrtc/modules/audio_processing/ |
| D | audio_processing_impl.h | 55 ChannelLayout output_layout,
73 ChannelLayout output_layout,
|
| D | audio_processing_impl.cc | 297 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/client/ |
| D | client.cc | 185 const Layout* output_layout) const { in ComputeConstant()
188 if (output_layout != nullptr) { in ComputeConstant()
189 *request.mutable_output_layout() = output_layout->ToProto(); in ComputeConstant()
|
| D | client.h | 193 const Layout* output_layout = nullptr) const;
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_output_layout_utils.cc | 65 const TPUEmbeddingOutputLayout& layout = config.output_layout(); in ComputeOutputTensorShapes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | service.cc | 1088 absl::optional<Layout> output_layout; in ComputeConstantGraph() local
1090 output_layout = Layout::CreateFromProto(arg->output_layout()); in ComputeConstantGraph()
1092 *output_layout, program_shape.result())); in ComputeConstantGraph()
1111 if (output_layout.has_value()) { in ComputeConstantGraph()
1112 result_literal = result_literal.Relayout(*output_layout); in ComputeConstantGraph()
|
| D | layout_assignment.h | 363 const Layout& output_layout, const HloInstruction* instruction,
|
| D | layout_assignment.cc | 988 const Layout& output_layout, const HloInstruction* instruction, in ChooseOperandLayoutFromOutputLayout() argument
1003 return absl::make_unique<Layout>(output_layout); in ChooseOperandLayoutFromOutputLayout()
1021 LayoutUtil::MinorToMajor(output_layout)); in ChooseOperandLayoutFromOutputLayout()
1040 int64 output_dim = LayoutUtil::Minor(output_layout, i); in ChooseOperandLayoutFromOutputLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1359 std::vector<int64> output_layout; in AlignLayouts() local
1360 output_layout.reserve(output_rank); in AlignLayouts()
1395 output_layout.push_back(j); in AlignLayouts()
1398 CHECK_EQ(output_layout.size(), output_rank); in AlignLayouts()
1401 output_layout); in AlignLayouts()
|
| D | shape_util_test.cc | 645 Layout output_layout = output_is_row_major in TEST() local
668 F32, {6, 2}, AsInt64Slice(output_layout.minor_to_major()))), in TEST()
|
| D | xla.proto | 447 LayoutProto output_layout = 2; field
|
| /external/webrtc/webrtc/modules/audio_processing/test/ |
| D | process_test.cc | 582 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/tensorflow/tensorflow/core/kernels/ |
| D | mkl_fused_ops_test.cc | 436 const Tensor& output_layout = *GetOutput(2); in Run() local
438 conv_comp.ConvertAndCompare(dtype, output, output_layout, expected); in Run()
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | tpu_embedding_configuration.proto | 94 TPUEmbeddingOutputLayout output_layout = 8; field
|
| /external/webrtc/talk/media/webrtc/ |
| D | fakewebrtcvoiceengine.h | 68 webrtc::AudioProcessing::ChannelLayout output_layout,
92 webrtc::AudioProcessing::ChannelLayout output_layout,
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 816 dnnLayout_t output_layout = static_cast<dnnLayout_t>(mkl_shape.GetTfLayout()); in ConvertMklToTF() local
821 mkl_shape.GetConvertedFlatData(output_layout, input_buffer, output_buffer); in ConvertMklToTF()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2406 const Layout& output_layout = output_shape.layout(); in EmitFastConcatenate() local
2407 auto output_min2maj = LayoutUtil::MinorToMajor(output_layout); in EmitFastConcatenate()
|