Home
last modified time | relevance | path

Searched refs:shape_indices (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Ddynamic_parameter_binding_invalid.mlir13 // Test missing `shape_indices` attribute in `mhlo.padding_map`.
19 // CHECK: requires 'shape_indices' array attribute in 'mhlo.padding_map' dict at arg 1
23 // Test bad `shape_indices` attribute type in `mhlo.padding_map`.
25 func @main(%arg0: tensor<i32>, %arg1: tensor<10xf32> {mhlo.padding_map = {shape_indices = ""}}) {
29 // CHECK: requires 'shape_indices' array attribute in 'mhlo.padding_map' dict at arg 1
35 func @main(%arg0: tensor<i32>, %arg1: tensor<10xf32> {mhlo.padding_map = {shape_indices = []}}) {
45 func @main(%arg0: tensor<i32>, %arg1: tensor<10xf32> {mhlo.padding_map = {shape_indices = [], paddi…
53 // Test mismatched `shape_indices` and `padding_arg_indices` lengths.
55 func @main(%arg0: tensor<i32>, %arg1: tensor<10xf32> {mhlo.padding_map = {shape_indices = [ 0: i32 …
59 // CHECK: requires 'shape_indices' and 'padding_arg_indices' array attributes in 'mhlo.padding_map'…
[all …]
Ddynamic_parameter_binding.mlir22 func @main(%arg0: tensor<10xf32> {mhlo.padding_map = {shape_indices = [0 : i32], padding_arg_indice…
45 func @main(%arg0: tensor<8x10xf32> {mhlo.padding_map = {shape_indices = [0 : i32, 1 : i32], padding…
78 …shape_indices = [0 : i32, 1 : i32], padding_arg_indices = [1 : i32, 2 : i32]}}, %arg1: tensor<i32>…
/external/tensorflow/tensorflow/compiler/xla/service/
Dallocation_tracker.cc70 std::vector<ShapeIndex> shape_indices; in RegisterInternal() local
74 shape_indices.push_back(index); in RegisterInternal()
78 for (const ShapeIndex& index : shape_indices) { in RegisterInternal()
103 std::vector<ShapeIndex> shape_indices; in Unregister() local
106 [&shape_indices](const Shape& /*subshape*/, const ShapeIndex& index) { in Unregister()
107 shape_indices.push_back(index); in Unregister()
109 for (const ShapeIndex& index : shape_indices) { in Unregister()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtpu_dynamic_padding_mapper.mlir20 …+}}: tensor<i1> {mhlo.padding_map = {padding_arg_indices = [1 : i32], shape_indices = [2 : i32]}},…
47 …or<i1> {mhlo.padding_map = {padding_arg_indices = [1 : i32, 2 : i32], shape_indices = [2 : i32, 3 …
79 …shape_indices = [2 : i32, 3 : i32]}}, %{{[a-z0-9]+}}: tensor<i1>, %{{[a-z0-9]+}}: tensor<i1>, %{{[…
100 …]+}}: tensor<i1> {mhlo.padding_map = {padding_arg_indices = [0 : i32], shape_indices = [2 : i32]}})
199 …+}}: tensor<i1> {mhlo.padding_map = {padding_arg_indices = [2 : i32], shape_indices = [2 : i32]}},…
243 …]+}}: tensor<i1> {mhlo.padding_map = {padding_arg_indices = [0 : i32], shape_indices = [6 : i32]}})
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_dynamic_padding_mapper.cc170 auto shape_indices = builder.getNamedAttr( in AnnotateFunctionArgumentsWithPaddings() local
177 builder.getDictionaryAttr({shape_indices, padding_arg_indices})); in AnnotateFunctionArgumentsWithPaddings()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1679 auto shape_indices = in AddDynamicParameterBindings() local
1681 if (!shape_indices) in AddDynamicParameterBindings()
1691 if (shape_indices.size() != padding_arg_indices.size()) in AddDynamicParameterBindings()
1693 i, shape_indices.size(), padding_arg_indices.size())); in AddDynamicParameterBindings()
1699 shape_indices.getValue(), padding_arg_indices.getValue()))) { in AddDynamicParameterBindings()