| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | stream_executor_util.cc | 95 std::vector<int64> output_layout; in StreamExecutorConvLayoutsToXlaLayouts() local
98 output_layout.push_back(dnums.output_batch_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
99 output_layout.push_back(dnums.output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
100 output_layout.insert(output_layout.end(), in StreamExecutorConvLayoutsToXlaLayouts()
105 output_layout.push_back(dnums.output_batch_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
106 output_layout.insert(output_layout.end(), in StreamExecutorConvLayoutsToXlaLayouts()
109 output_layout.push_back(dnums.output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
119 LayoutUtil::MakeLayoutFromMajorToMinor(output_layout)); in StreamExecutorConvLayoutsToXlaLayouts()
162 DataLayout output_layout; in XlaConvLayoutsToStreamExecutorLayouts() local
164 output_layout = DataLayout::kBatchDepthYX; in XlaConvLayoutsToStreamExecutorLayouts()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | grouped_convolution_test.cc | 42 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 | 40 std::vector<int64> output_layout; member
74 config.output_layout = {2, 3, 0, 1}; in GetConv2DTestCases()
91 config.output_layout = {2, 3, 0, 1}; in GetConv2DTestCases()
118 absl::StrJoin(spec.output_layout, "_")); in BatchGroupedConvolution2DTestDataToString()
145 absl::StrJoin(spec.output_layout, ","), data_type, in BuildHloTextBatchGroupedConvolution2D()
|
| D | conv_depthwise_common.cc | 45 absl::StrJoin(spec.output_layout, "_"), data_type); in DepthwiseConvolution2DTestDataToString()
78 absl::StrJoin(spec.output_layout, ","), data_type, in BuildHloTextDepthwiseConvolution2D()
102 absl::StrJoin(spec.output_layout, ","), data_type, in BuildHloTextDepthwiseConvolution2D()
126 absl::StrJoin(spec.output_layout, ","), data_type, in BuildHloTextDepthwiseConvolution2D()
|
| D | conv_depthwise_common.h | 41 std::vector<int64> output_layout; member
|
| D | compute_constant_test.cc | 74 Layout* output_layout = nullptr) { in ComputeConstantLiteral() argument
77 client->ComputeConstant(subgraph, output_layout)); in ComputeConstantLiteral()
|
| D | conv_depthwise_test.cc | 60 config.output_layout = {3, 0, 2, 1}; in GetConv2DTestCases()
|
| D | reduce_window_test.cc | 392 std::vector<int64> output_layout = {1, 5, 3, 2, 0, 4}; in XLA_TEST_P() local
395 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 | 1109 absl::optional<Layout> output_layout; in ComputeConstantGraph() local
1111 output_layout = Layout::CreateFromProto(arg->output_layout()); in ComputeConstantGraph()
1113 *output_layout, program_shape.result())); in ComputeConstantGraph()
1132 if (output_layout.has_value()) { in ComputeConstantGraph()
1133 result_literal = result_literal.Relayout(*output_layout); in ComputeConstantGraph()
|
| D | layout_assignment.h | 368 const Layout& output_layout, const HloInstruction* instruction,
|
| D | layout_assignment.cc | 1044 const Layout& output_layout, const HloInstruction* instruction, in ChooseOperandLayoutFromOutputLayout() argument
1059 return absl::make_unique<Layout>(output_layout); in ChooseOperandLayoutFromOutputLayout()
1078 LayoutUtil::MinorToMajor(output_layout)); in ChooseOperandLayoutFromOutputLayout()
1097 int64 output_dim = LayoutUtil::Minor(output_layout, i); in ChooseOperandLayoutFromOutputLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1386 std::vector<int64> output_layout; in AlignLayouts() local
1387 output_layout.reserve(output_rank); in AlignLayouts()
1422 output_layout.push_back(j); in AlignLayouts()
1425 CHECK_EQ(output_layout.size(), output_rank); in AlignLayouts()
1428 output_layout); in AlignLayouts()
|
| D | shape_util_test.cc | 667 Layout output_layout = output_is_row_major in TEST() local
690 F32, {6, 2}, AsInt64Slice(output_layout.minor_to_major()))), in TEST()
|
| D | xla.proto | 454 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 | 581 const Tensor& output_layout = *GetOutput(2); in Run() local
583 conv_comp.ConvertAndCompare(dtype, output, output_layout, expected); in Run()
828 const Tensor& output_layout = *GetOutput(3); in Run() local
830 conv_comp.ConvertAndCompareIntegral(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,
|