Searched defs:window_dimensions (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | padding.cc | 27 absl::Span<const int64> window_dimensions, in ValidatePaddingValues() 43 absl::Span<const int64> window_dimensions, in MakePadding()
|
D | xla_builder.cc | 1139 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/ |
D | reduce_window_op.cc | 41 std::vector<int64> window_dimensions; in Compile() local
|
D | xla_select_and_scatter_op.cc | 42 std::vector<int64> window_dimensions; in Compile() local
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | reduce_window_test.cc | 73 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()
|
D | select_and_scatter_test.cc | 45 absl::Span<const int64> window_dimensions; member
|
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | pooling.cc | 109 absl::Span<const int64> window_dimensions, in AvgPoolDivideByCount()
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | local_computation_builder.cc | 552 absl::Span<const int64> window_dimensions, in SelectAndScatterWithGeneralPadding() 667 absl::Span<const int64> window_dimensions, in ReduceWindowWithGeneralPadding()
|
D | xla_client.py | 1282 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/ |
D | xla.py | 353 window_dimensions, argument
|