Home
last modified time | relevance | path

Searched refs:output_shapes (Results 1 – 25 of 624) sorted by relevance

12345678910>>...25

/external/tensorflow/tensorflow/lite/kernels/ctc/
Dctc_beam_search_decoder_test.cc84 std::vector<std::vector<int>> output_shapes; in GetOutputShapes() local
86 output_shapes.push_back(GetTensorShape(output)); in GetOutputShapes()
88 return output_shapes; in GetOutputShapes()
105 const std::vector<std::vector<int>>& output_shapes = m.GetOutputShapes(); in TEST() local
106 EXPECT_EQ(output_shapes.size(), 4); in TEST()
107 EXPECT_THAT(output_shapes[0], ElementsAre(1, 2)); in TEST()
108 EXPECT_THAT(output_shapes[1], ElementsAre(1)); in TEST()
109 EXPECT_THAT(output_shapes[2], ElementsAre(2)); in TEST()
110 EXPECT_THAT(output_shapes[3], ElementsAre(1, 1)); in TEST()
137 const std::vector<std::vector<int>>& output_shapes = m.GetOutputShapes(); in TEST() local
[all …]
/external/tensorflow/tensorflow/core/ops/
Dfunctional_ops.cc94 std::vector<PartialTensorShape> output_shapes; in IfShapeInferenceFn() local
95 TF_RETURN_IF_ERROR(c->GetAttr("output_shapes", &output_shapes)); in IfShapeInferenceFn()
98 if (output_shapes.empty()) return shape_inference::UnknownShape(c); in IfShapeInferenceFn()
99 if (output_shapes.size() != c->num_outputs()) { in IfShapeInferenceFn()
102 output_shapes.size(), " vs. ", c->num_outputs()); in IfShapeInferenceFn()
104 for (size_t i = 0; i < output_shapes.size(); ++i) { in IfShapeInferenceFn()
107 output_shapes[i], &output_shape_handle)); in IfShapeInferenceFn()
139 std::vector<PartialTensorShape> output_shapes; in CaseShapeInferenceFn() local
140 TF_RETURN_IF_ERROR(c->GetAttr("output_shapes", &output_shapes)); in CaseShapeInferenceFn()
143 if (output_shapes.empty()) return shape_inference::UnknownShape(c); in CaseShapeInferenceFn()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dprelu.cc48 if (alpha->shape.v != ctx.output_shapes[0][3]) { in GenerateCode()
60 uint3(static_cast<int>(ctx.output_shapes[0][2]), in GenerateCode()
61 static_cast<int>(ctx.output_shapes[0][1]), in GenerateCode()
62 DivideRoundUp(static_cast<int>(ctx.output_shapes[0][3]), 4)), in GenerateCode()
83 if (alpha->shape.h != ctx.output_shapes[0][1] || in GenerateCode()
84 alpha->shape.w != ctx.output_shapes[0][2] || in GenerateCode()
85 alpha->shape.c != ctx.output_shapes[0][3]) { in GenerateCode()
91 uint3(static_cast<int>(ctx.output_shapes[0][2]), in GenerateCode()
92 static_cast<int>(ctx.output_shapes[0][1]), in GenerateCode()
93 DivideRoundUp(static_cast<int>(ctx.output_shapes[0][3]), 4)); in GenerateCode()
[all …]
Dreshape.cc42 ctx.output_shapes[0][1] * ctx.output_shapes[0][2] * in GenerateCode()
43 ctx.output_shapes[0][3]) { in GenerateCode()
48 if (attr.new_shape.h != ctx.output_shapes[0][1] || in GenerateCode()
49 attr.new_shape.w != ctx.output_shapes[0][2] || in GenerateCode()
50 attr.new_shape.c != ctx.output_shapes[0][3]) { in GenerateCode()
76 {"output_data_0_w", static_cast<int>(ctx.output_shapes[0][2])}, in GenerateCode()
77 {"output_channels", static_cast<int>(ctx.output_shapes[0][3])}, in GenerateCode()
Dresize.cc43 if (ctx.input_shapes[0][2] > ctx.output_shapes[0][2] || in GenerateCode()
44 ctx.input_shapes[0][1] > ctx.output_shapes[0][1]) { in GenerateCode()
47 if (ctx.output_shapes[0][2] != attr.new_shape.w || in GenerateCode()
48 ctx.output_shapes[0][1] != attr.new_shape.h) { in GenerateCode()
52 if (ctx.input_shapes[0][3] != ctx.output_shapes[0][3]) { in GenerateCode()
74 ctx.output_shapes[0][2], attr), in GenerateCode()
76 ctx.output_shapes[0][1], attr))}, in GenerateCode()
Dsoftmax.cc48 if (ctx.input_shapes[0] != ctx.output_shapes[0]) { in GenerateCode()
64 const int depth = DivideRoundUp(ctx.output_shapes[0][3], 4); in GenerateCodeFor1x1()
70 {"mask", GetMask(ctx.output_shapes[0][3])}, in GenerateCodeFor1x1()
170 DivideRoundUp(static_cast<int>(ctx.output_shapes[0][3]), 4)}, in GenerateCodeGeneral()
171 {"mask", GetMask(ctx.output_shapes[0][3])}, in GenerateCodeGeneral()
204 uint3(static_cast<int>(ctx.output_shapes[0][2]), in GenerateCodeGeneral()
205 static_cast<int>(ctx.output_shapes[0][1]), 1), in GenerateCodeGeneral()
/external/tensorflow/tensorflow/python/data/ops/
Diterator_ops.py102 output_shapes, output_classes): argument
128 if (output_types is None or output_shapes is None
137 output_types, output_shapes, output_classes)
150 output_shapes=None, argument
219 if output_shapes is None:
220 output_shapes = nest.map_structure(
223 output_shapes = nest.map_structure_up_to(output_types,
225 output_shapes)
228 nest.assert_same_structure(output_types, output_shapes)
230 output_types, output_shapes, output_classes)
[all …]
/external/tensorflow/tensorflow/core/transforms/consolidate_attrs/tests/
Dcf_op_output_shapes.mlir7 // CHECK-NOT: output_shapes
10 Tcond = i1, Tin = [i32], Tout = [i32], output_shapes = [#tf_type.shape<4>],
21 // CHECK-NOT: output_shapes
24 Tin = [i32], Tout = [i32], output_shapes = [#tf_type.shape<4>],
35 // CHECK-NOT: output_shapes
38 T = [i32], output_shapes = [#tf_type.shape<4>],
46 // `output_shapes` is an optional attribute in TF. When imported to TFG, if the
53 // CHECK-SAME: output_shapes = []
56 T = [i32], output_shapes = [],
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dmap_defun.py25 output_shapes, argument
54 if not isinstance(output_shapes, list):
62 output_shapes = [tensor_shape.TensorShape(s) for s in output_shapes]
64 output_dtypes, output_shapes, concrete_fn,
/external/tensorflow/tensorflow/compiler/xla/service/
Dreduce_decomposer.cc87 std::vector<Shape> output_shapes; in HandleReduce() local
90 output_shapes.push_back(ShapeUtil::GetTupleElementShape(shape, i)); in HandleReduce()
91 TF_RET_CHECK(output_shapes[i].layout() == output_shapes[0].layout()); in HandleReduce()
94 output_shapes.push_back(shape); in HandleReduce()
97 TF_RET_CHECK(!output_shapes.empty()); in HandleReduce()
99 ShapeUtil::MakeMaybeTupleShape(output_shapes)) { in HandleReduce()
116 auto copy = MakeCopyHlo(from, output_shapes[i]); in HandleReduce()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/ops/
Dtrt_engine_op.cc49 std::vector<tensorflow::PartialTensorShape> output_shapes; in __anonb035d4020102() local
50 TF_RETURN_IF_ERROR(c->GetAttr("output_shapes", &output_shapes)); in __anonb035d4020102()
52 for (int i = 0; i < output_shapes.size(); i++) { in __anonb035d4020102()
56 output_shapes[i], &output_shape_handle)); in __anonb035d4020102()
/external/tensorflow/tensorflow/core/ir/tests/
Dconcrete-ops.mlir59 // CHECK-SAME: output_shapes = [#tf_type.shape<>],
60 output_shapes = [#tf_type.shape<>],
75 // CHECK-SAME: output_shapes = [#tf_type.shape<>, #tf_type.shape<>],
76 output_shapes = [#tf_type.shape<>, #tf_type.shape<>],
90 // CHECK-SAME: output_shapes = [#tf_type.shape<>, #tf_type.shape<>]}
96 output_shapes = [#tf_type.shape<>, #tf_type.shape<>]
114 // CHECK-SAME: output_shapes = [#tf_type.shape<>],
115 output_shapes = [#tf_type.shape<>],
125 // CHECK-SAME: output_shapes = [#tf_type.shape<>]}
131 output_shapes = [#tf_type.shape<>]
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util_test.cc45 std::vector<PartialTensorShape> output_shapes; in TEST() local
47 &output_shapes)); in TEST()
48 EXPECT_EQ(output_shapes.size(), 1); in TEST()
50 output_shapes[0].AsProto(&shape_proto); in TEST()
/external/tensorflow/tensorflow/core/transforms/functional_to_region/tests/
Dif.mlir11 {output_shapes = [#tf_type.shape<>, #tf_type.shape<>]}
13 %ctl_0 = NoOp [%ctl] name("NoOp") {output_shapes = []}
15 {output_shapes = [#tf_type.shape<>]}
18 {output_shapes = [#tf_type.shape<>]}
33 {output_shapes = [#tf_type.shape<>]}
36 {output_shapes = [#tf_type.shape<>]}
65 output_shapes = [#tf_type.shape<>, #tf_type.shape<>],
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DModelDataset.pbtxt18 name: "output_shapes"
55 name: "output_shapes"
99 name: "output_shapes"
143 name: "output_shapes"
194 name: "output_shapes"
258 name: "output_shapes"
DTensorSliceDataset.pbtxt18 name: "output_shapes"
42 name: "output_shapes"
73 name: "output_shapes"
118 name: "output_shapes"
176 name: "output_shapes"
234 name: "output_shapes"
DPrefetchDataset.pbtxt22 name: "output_shapes"
50 name: "output_shapes"
77 name: "output_shapes"
118 name: "output_shapes"
166 name: "output_shapes"
228 name: "output_shapes"
303 name: "output_shapes"
DSkipDataset.pbtxt22 name: "output_shapes"
50 name: "output_shapes"
77 name: "output_shapes"
118 name: "output_shapes"
172 name: "output_shapes"
DTakeDataset.pbtxt22 name: "output_shapes"
50 name: "output_shapes"
77 name: "output_shapes"
118 name: "output_shapes"
172 name: "output_shapes"
DBatchDataset.pbtxt22 name: "output_shapes"
50 name: "output_shapes"
77 name: "output_shapes"
118 name: "output_shapes"
172 name: "output_shapes"
DConcatenateDataset.pbtxt22 name: "output_shapes"
50 name: "output_shapes"
77 name: "output_shapes"
118 name: "output_shapes"
172 name: "output_shapes"
DCacheDataset.pbtxt22 name: "output_shapes"
50 name: "output_shapes"
77 name: "output_shapes"
118 name: "output_shapes"
172 name: "output_shapes"
DRepeatDataset.pbtxt22 name: "output_shapes"
50 name: "output_shapes"
77 name: "output_shapes"
118 name: "output_shapes"
172 name: "output_shapes"
DMapDataset.pbtxt31 name: "output_shapes"
68 name: "output_shapes"
104 name: "output_shapes"
147 name: "output_shapes"
197 name: "output_shapes"
261 name: "output_shapes"
338 name: "output_shapes"
/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/
Dfunc_attributes.mlir9 …%3 = "tf.RangeDataset"(%0, %1, %2) {device = "/device:CPU:0", output_shapes = [#tf_type.shape<>], …
10 …:CPU:0") "tf.FlatMapDataset"({{.*}}) {Targuments = [], metadata = "", output_shapes = [#corert.sha…
11 …vice = "/device:CPU:0", f = @__inference_Dataset_flat_map_lambda_190, output_shapes = [#tf_type.sh…
19 …%3 = "tf.RangeDataset"(%0, %2, %1) {device = "/device:CPU:0", output_shapes = [#tf_type.shape<>], …
20 …vice:CPU:0") "tf.MapDataset"({{.*}}) {Targuments = [], metadata = "", output_shapes = [#corert.sha…
21 … f = @__inference_Dataset_map_lambda_160, f._tf_data_function = true, output_shapes = [#tf_type.sh…

12345678910>>...25