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 | 1418 std::vector<int64> window_dimensions( in ConvWithGeneralDimensions() local
1462 std::vector<int64> window_dimensions( in ConvGeneralDilated() local
1496 std::vector<int64> window_dimensions( in DynamicConvInstruction() local
2489 absl::Span<const int64> window_dimensions, in ReduceWindow()
2500 absl::Span<const int64> window_dimensions, in ReduceWindow()
2552 absl::Span<const int64> window_dimensions, in ReduceWindowWithGeneralPadding()
2597 absl::Span<const int64> window_dimensions, in ReduceWindowInternal()
3027 absl::Span<const int64> window_dimensions, in SelectAndScatter()
3073 absl::Span<const int64> window_dimensions, in SelectAndScatterInternal()
3103 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 | 1109 std::vector<int64> window_dimensions = {1, 2, 4}; in TEST_F() local
1171 std::vector<int64> window_dimensions = {1, 2, 4}; in TEST_F() local
|
| D | shape_inference.cc | 2349 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/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/xla/client/lib/ |
| D | pooling.cc | 110 absl::Span<const int64> window_dimensions, in AvgPoolDivideByCount()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 450 llvm::SmallVector<int64_t, 4> window_strides, window_dimensions; in ImportInstructionImpl() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 351 window_dimensions, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2288 mlir::DenseIntElementsAttr window_dimensions = in EmitSelectAndScatterFromMlir() local
|