Home
last modified time | relevance | path

Searched defs:window_dimensions (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dpadding.cc27 absl::Span<const int64> window_dimensions, in ValidatePaddingValues()
43 absl::Span<const int64> window_dimensions, in MakePadding()
Dxla_builder.cc1139 std::vector<int64> window_dimensions( in ConvWithGeneralDimensions() local
1180 std::vector<int64> window_dimensions( in ConvGeneralDilated() local
1211 absl::Span<const int64> window_dimensions, in MakeWindow()
1995 absl::Span<const int64> window_dimensions, in ReduceWindow()
2018 absl::Span<const int64> window_dimensions, in ReduceWindowWithGeneralPadding()
2246 absl::Span<const int64> window_dimensions, in SelectAndScatter()
2263 absl::Span<const int64> window_dimensions, in SelectAndScatterWithGeneralPadding()
3212 absl::Span<const int64> window_dimensions, in ReduceWindow()
3222 absl::Span<const int64> window_dimensions, in ReduceWindowWithGeneralPadding()
3260 absl::Span<const int64> window_dimensions, in SelectAndScatter()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreduce_window_op.cc41 std::vector<int64> window_dimensions; in Compile() local
Dxla_select_and_scatter_op.cc42 std::vector<int64> window_dimensions; in Compile() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_window_test.cc73 absl::Span<const int64> window_dimensions, in ReduceWindowAdd()
84 absl::Span<const int64> window_dimensions, in ReduceWindowMax()
95 absl::Span<const int64> window_dimensions, in ReduceWindowMin()
Dselect_and_scatter_test.cc45 absl::Span<const int64> window_dimensions; member
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc109 absl::Span<const int64> window_dimensions, in AvgPoolDivideByCount()
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc552 absl::Span<const int64> window_dimensions, in SelectAndScatterWithGeneralPadding()
667 absl::Span<const int64> window_dimensions, in ReduceWindowWithGeneralPadding()
Dxla_client.py1282 def SelectAndScatter(self, operand, select, window_dimensions, window_strides, argument
1473 window_dimensions, window_strides, padding): argument
1495 self, operand, init_value, computation_to_apply, window_dimensions, argument
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py353 window_dimensions, argument