Searched defs:window_dimensions (Results 1 – 13 of 13) 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 | 1430 std::vector<int64> window_dimensions( in ConvWithGeneralDimensions() local
1474 std::vector<int64> window_dimensions( in ConvGeneralDilated() local
1508 std::vector<int64> window_dimensions( in DynamicConvInstruction() local
2517 absl::Span<const int64> window_dimensions, in ReduceWindow()
2528 absl::Span<const int64> window_dimensions, in ReduceWindow()
2580 absl::Span<const int64> window_dimensions, in ReduceWindowWithGeneralPadding()
2625 absl::Span<const int64> window_dimensions, in ReduceWindowInternal()
3116 absl::Span<const int64> window_dimensions, in SelectAndScatter()
3162 absl::Span<const int64> window_dimensions, in SelectAndScatterInternal()
3192 absl::Span<const int64> window_dimensions, in SelectAndScatterWithGeneralPadding()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tree_reduction_rewriter.cc | 53 std::vector<int64> window_dimensions; in HandleReduce() local
|
| D | shape_inference_test.cc | 1130 std::vector<int64> window_dimensions = {1, 2, 4}; in TEST_F() local
1192 std::vector<int64> window_dimensions = {1, 2, 4}; in TEST_F() local
|
| D | shape_inference.cc | 2463 absl::Span<const int64> window_dimensions, in InferWindowFromDimensions()
|
| /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/client/lib/ |
| D | pooling.cc | 110 absl::Span<const int64> window_dimensions, in AvgPoolDivideByCount()
|
| /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 std::vector<int64> window_dimensions; member
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 462 llvm::SmallVector<int64_t, 4> window_strides, window_dimensions; in ImportInstructionImpl() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 413 window_dimensions, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2157 mlir::DenseIntElementsAttr window_dimensions = in EmitSelectAndScatter() local
|