Home
last modified time | relevance | path

Searched refs:Tile (Results 1 – 25 of 68) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/
Dlayout.h33 class Tile {
35 Tile() = default;
36 explicit Tile(absl::Span<const int64> dimensions) in Tile() function
40 static Tile CreateFromProto(const TileProto& tile_proto) { in CreateFromProto()
41 return Tile(AsInt64Slice(tile_proto.dimensions())); in CreateFromProto()
45 bool operator==(const Tile& other) const {
48 bool operator!=(const Tile& other) const { return !(*this == other); }
58 Tile& add_dimensions(int64 value) { in add_dimensions()
63 Tile& clear_dimensions() { in clear_dimensions()
88 Layout(absl::Span<const int64> minor_to_major, absl::Span<const Tile> tiles,
[all …]
Dlayout_test.cc40 EXPECT_EQ(Layout({3, 2, 1, 0}, {Tile({42, 123}), Tile({4, 5})}).ToString(), in TEST_F()
43 Layout({1, 0}, {Tile({2, 55})}).set_element_size_in_bits(42).ToString(), in TEST_F()
46 Layout({1, 0}, {Tile({-2, 55})}).set_element_size_in_bits(42).ToString(), in TEST_F()
53 oss << Tile({7, 8}); in TEST_F()
76 EXPECT_EQ(Layout({0, 1, 2}, {Tile({42, 44})}), in TEST_F()
77 Layout({0, 1, 2}, {Tile({42, 44})})); in TEST_F()
78 EXPECT_NE(Layout({0, 1, 2}, {Tile({42, 44})}), in TEST_F()
79 Layout({0, 1, 2}, {Tile({42, 45})})); in TEST_F()
80 EXPECT_NE(Layout({0, 1, 2}, {Tile({42, 44})}), Layout({0, 1, 2, 3})); in TEST_F()
92 Layout::Equal()(Layout({0, 1, 2}, {Tile({42, 44})}), Layout({0, 1, 2}))); in TEST_F()
[all …]
Dlayout.cc24 TileProto Tile::ToProto() const { in ToProto()
32 string Tile::ToString() const { in ToString()
57 *layout.add_tiles() = Tile::CreateFromProto(tile_proto); in CreateFromProto()
71 for (const Tile& tile : tiles()) { in ToProto()
84 for (Tile tile : tiles()) { in ToString()
118 std::ostream& operator<<(std::ostream& out, const Tile& tile) { in operator <<()
Dlayout_util.cc57 absl::Span<const int64> minor_to_major, absl::Span<const Tile> tiles, in MakeLayout()
64 for (Tile tile : tiles) { in MakeLayout()
66 if (dim < 0 && dim != Tile::kCombineDimension) { in MakeLayout()
463 for (Tile tile : layout.tiles()) { in Hash()
/external/robolectric-shadows/shadows/framework/src/main/java/org/robolectric/shadows/
DShadowTileService.java4 import android.service.quicksettings.Tile;
13 private Tile tile;
16 protected final Tile getQsTile() { in getQsTile()
23 private static Tile createTile() { in createTile()
24 return Shadow.newInstanceOf(Tile.class); in createTile()
DShadowTile.java4 import android.service.quicksettings.Tile;
8 @Implements(value = Tile.class, minSdk = Build.VERSION_CODES.N)
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_test.cc78 TEST_F(HloShardingTest, Tile) { in TEST_F() argument
81 HloSharding sharding = HloSharding::Tile(MakeArray({2, 2}, {0, 0, 2, 3})); in TEST_F()
88 HloSharding sharding = HloSharding::Tile(MakeArray({2, 2}, {0, 1, 2, 3})); in TEST_F()
96 HloSharding sharding = HloSharding::Tile(MakeArray({2, 2}, {0, 3, 2, 1})); in TEST_F()
133 HloSharding tiled_sharding = HloSharding::Tile(Array<int64>({{0, 1}})); in TEST_F()
184 HloSharding sharding1 = HloSharding::Tile(MakeArray({2, 2}, {0, 3, 2, 1})); in TEST_F()
185 HloSharding sharding2 = HloSharding::Tile(MakeArray({2, 2}, {0, 3, 2, 1})); in TEST_F()
247 HloSharding::Tile(Array3D<int64>({{{2, 3}}, {{5, 7}}})); in TEST_F()
256 {HloSharding::Replicate(), HloSharding::Tile(Array2D<int64>({{3, 5}})), in TEST_F()
264 HloSharding::Tile(Array4D<int64>({{{{0, 1}, {2, 3}}}})); in TEST_F()
[all …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_tile.cc72 inline void Tile(const Array& input_array, const Array& multiples_array, in Tile() function
139 Tile<ArrayDataType::kFloat>(input_array, multiples_array, &output_array); in Run()
142 Tile<ArrayDataType::kUint8>(input_array, multiples_array, &output_array); in Run()
145 Tile<ArrayDataType::kInt16>(input_array, multiples_array, &output_array); in Run()
148 Tile<ArrayDataType::kInt32>(input_array, multiples_array, &output_array); in Run()
151 Tile<ArrayDataType::kInt64>(input_array, multiples_array, &output_array); in Run()
154 Tile<ArrayDataType::kComplex64>(input_array, multiples_array, in Run()
/external/tensorflow/tensorflow/core/kernels/
Dtile_functor_cpu.cc55 template struct Tile<CPUDevice, T, int32>; \
56 template struct Tile<CPUDevice, T, int64>;
77 template struct Tile<SYCLDevice, T, int32>; \
78 template struct Tile<SYCLDevice, T, int64>;
Dtile_functor_gpu_float.cu.cc26 template struct Tile<GpuDevice, float, int32>; variable
27 template struct Tile<GpuDevice, float, int64>; variable
Dtile_functor_gpu_double.cu.cc26 template struct Tile<GpuDevice, double, int32>; variable
27 template struct Tile<GpuDevice, double, int64>; variable
Dtile_functor_gpu_bool.cu.cc26 template struct Tile<GpuDevice, bool, int32>; variable
27 template struct Tile<GpuDevice, bool, int64>; variable
Dtile_functor_gpu_int64.cu.cc26 template struct Tile<GpuDevice, int64, int32>; variable
27 template struct Tile<GpuDevice, int64, int64>; variable
Dtile_functor_gpu_complex128.cu.cc26 template struct Tile<GpuDevice, complex128, int32>; variable
27 template struct Tile<GpuDevice, complex128, int64>; variable
Dtile_functor_gpu_complex64.cu.cc26 template struct Tile<GpuDevice, complex64, int32>; variable
27 template struct Tile<GpuDevice, complex64, int64>; variable
Dtile_functor_gpu_int16.cu.cc26 template struct Tile<GpuDevice, int16, int32>; variable
27 template struct Tile<GpuDevice, int16, int64>; variable
Dtile_functor_gpu_int32.cu.cc26 template struct Tile<GpuDevice, int32, int32>; variable
27 template struct Tile<GpuDevice, int32, int64>; variable
Dtile_functor_gpu_half.cu.cc26 template struct Tile<GpuDevice, Eigen::half, int32>; variable
27 template struct Tile<GpuDevice, Eigen::half, int64>; variable
/external/tensorflow/tensorflow/lite/kernels/
Dtile.cc117 void Tile(const TfLiteIntArray& in_dimensions, const TfLiteTensor* in_data, in Tile() function
174 Tile<float>(*(input->dims), input, multipliers, output); in Eval()
177 Tile<uint8_t>(*(input->dims), input, multipliers, output); in Eval()
180 Tile<int32_t>(*(input->dims), input, multipliers, output); in Eval()
183 Tile<int64_t>(*(input->dims), input, multipliers, output); in Eval()
186 Tile<bool>(*(input->dims), input, multipliers, output); in Eval()
/external/robolectric-shadows/robolectric/src/test/java/org/robolectric/shadows/
DShadowTileTest.java6 import android.service.quicksettings.Tile;
19 private Tile tile;
24 tile = Shadow.newInstanceOf(Tile.class); in setUp()
DShadowTileServiceTest.java6 import android.service.quicksettings.Tile;
28 Tile tile = tileService.getQsTile(); in getTile()
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_Tile.pbtxt2 graph_op_name: "Tile"
4 name: "Tile"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TileGrad.pbtxt3 summary: "Returns the gradient of `Tile`."
5 Since `Tile` takes an input and repeats the input `multiples` times
/external/subsampling-scale-image-view/library/src/main/java/com/davemorrissey/labs/subscaleview/
DSubsamplingScaleImageView.java149 private Map<Integer, List<Tile>> tileMap;
532 for (Map.Entry<Integer, List<Tile>> tileMapEntry : tileMap.entrySet()) { in reset()
533 for (Tile tile : tileMapEntry.getValue()) { in reset()
1035 for (Map.Entry<Integer, List<Tile>> tileMapEntry : tileMap.entrySet()) { in onDraw()
1037 for (Tile tile : tileMapEntry.getValue()) { in onDraw()
1046 for (Map.Entry<Integer, List<Tile>> tileMapEntry : tileMap.entrySet()) { in onDraw()
1048 for (Tile tile : tileMapEntry.getValue()) { in onDraw()
1168 for (Map.Entry<Integer, List<Tile>> tileMapEntry : tileMap.entrySet()) { in isBaseLayerReady()
1170 for (Tile tile : tileMapEntry.getValue()) { in isBaseLayerReady()
1269 List<Tile> baseGrid = tileMap.get(fullImageSampleSize); in initialiseBaseLayer()
[all …]
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_Tile.pbtxt2 graph_op_name: "Tile"

123