| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_beam_search_decoder_test.cc | 84 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/lite/delegates/gpu/gl/kernels/ |
| D | prelu.cc | 45 if (alpha->shape.v != ctx.output_shapes[0][3]) { in GenerateCode()
70 uint3(static_cast<int>(ctx.output_shapes[0][2]), in GenerateCode()
71 static_cast<int>(ctx.output_shapes[0][1]), in GenerateCode()
72 DivideRoundUp(static_cast<int>(ctx.output_shapes[0][3]), in GenerateCode()
94 if (alpha->shape.h != ctx.output_shapes[0][1] || in GenerateCode()
95 alpha->shape.w != ctx.output_shapes[0][2] || in GenerateCode()
96 alpha->shape.c != ctx.output_shapes[0][3]) { in GenerateCode()
102 uint3(static_cast<int>(ctx.output_shapes[0][2]), in GenerateCode()
103 static_cast<int>(ctx.output_shapes[0][1]), in GenerateCode()
104 DivideRoundUp(static_cast<int>(ctx.output_shapes[0][3]), 4)); in GenerateCode()
[all …]
|
| D | reshape.cc | 39 ctx.output_shapes[0][1] * ctx.output_shapes[0][2] * in GenerateCode()
40 ctx.output_shapes[0][3]) { in GenerateCode()
45 if (attr.new_shape.h != ctx.output_shapes[0][1] || in GenerateCode()
46 attr.new_shape.w != ctx.output_shapes[0][2] || in GenerateCode()
47 attr.new_shape.c != ctx.output_shapes[0][3]) { in GenerateCode()
73 {"output_data_0_w", static_cast<int>(ctx.output_shapes[0][2])}, in GenerateCode()
74 {"output_channels", static_cast<int>(ctx.output_shapes[0][3])}, in GenerateCode()
|
| D | resize.cc | 40 if (ctx.input_shapes[0][2] > ctx.output_shapes[0][2] || in GenerateCode()
41 ctx.input_shapes[0][1] > ctx.output_shapes[0][1]) { in GenerateCode()
44 if (ctx.output_shapes[0][2] != attr.new_shape.w || in GenerateCode()
45 ctx.output_shapes[0][1] != attr.new_shape.h) { in GenerateCode()
49 if (ctx.input_shapes[0][3] != ctx.output_shapes[0][3]) { in GenerateCode()
71 ctx.output_shapes[0][2], attr), in GenerateCode()
73 ctx.output_shapes[0][1], attr))}, in GenerateCode()
|
| D | softmax.cc | 47 if (ctx.input_shapes[0] != ctx.output_shapes[0]) { in GenerateCode()
63 const int depth = DivideRoundUp(ctx.output_shapes[0][3], 4); in GenerateCodeFor1x1()
69 {"mask", GetMask(ctx.output_shapes[0][3])}, in GenerateCodeFor1x1()
169 DivideRoundUp(static_cast<int>(ctx.output_shapes[0][3]), 4)}, in GenerateCodeGeneral()
170 {"mask", GetMask(ctx.output_shapes[0][3])}, in GenerateCodeGeneral()
203 uint3(static_cast<int>(ctx.output_shapes[0][2]), in GenerateCodeGeneral()
204 static_cast<int>(ctx.output_shapes[0][1]), 1), in GenerateCodeGeneral()
|
| D | mean.cc | 184 if (!(ctx.input_shapes.size() == 1 && ctx.output_shapes.size() == 1 && in GenerateCode()
185 ctx.output_shapes[0][1] == 1 && ctx.output_shapes[0][2] == 1 && in GenerateCode()
186 ctx.output_shapes[0][3] == ctx.input_shapes[0][3])) { in GenerateCode()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | functional_ops.cc | 94 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 …]
|
| D | tpu_embedding_ops.cc | 73 std::vector<TensorShapeProto> output_shapes; in __anon222eb4110102() local
74 TF_RETURN_IF_ERROR(ComputeOutputTensorShapes(config, &output_shapes)); in __anon222eb4110102()
75 if (c->num_outputs() != output_shapes.size()) { in __anon222eb4110102()
81 c->MakeShapeFromShapeProto(output_shapes[i], &output_shape)); in __anon222eb4110102()
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | iterator_ops.py | 84 output_shapes, output_classes): argument
106 if (output_types is None or output_shapes is None
112 output_types, output_shapes, output_classes)
125 output_shapes=None, argument
194 if output_shapes is None:
195 output_shapes = nest.map_structure(
198 output_shapes = nest.map_structure_up_to(output_types,
200 output_shapes)
203 nest.assert_same_structure(output_types, output_shapes)
205 output_types, output_shapes, output_classes)
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | map_defun.py | 29 output_shapes, argument
57 if not isinstance(output_shapes, list):
65 output_shapes = [tensor_shape.TensorShape(s) for s in output_shapes]
67 output_dtypes, output_shapes, concrete_fn,
|
| D | interleave_ops.py | 126 output_shapes = dataset_ops.get_legacy_output_shapes(self._data_inputs[0])
128 output_shapes = nest.pack_sequence_as(output_shapes, [
130 nest.flatten(output_shapes),
135 first_output_types, output_shapes, first_output_classes)
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util_test.cc | 45 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/kernels/data/ |
| D | window_dataset.cc | 32 std::vector<PartialTensorShape> output_shapes) in WindowDataset() argument
36 output_shapes_(std::move(output_shapes)) {}
46 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anonb5d693710111::WindowDataset
133 std::vector<PartialTensorShape> output_shapes, in NewWindowDataset() argument
138 std::move(output_shapes)); in NewWindowDataset()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.-iterator.pbtxt | 19 name: "output_shapes"
28 …\'self\', \'iterator_resource\', \'initializer\', \'output_types\', \'output_shapes\', \'output_cl…
32 …argspec: "args=[\'string_handle\', \'output_types\', \'output_shapes\', \'output_classes\'], varar…
36 …argspec: "args=[\'output_types\', \'output_shapes\', \'shared_name\', \'output_classes\'], varargs…
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | PrefetchDataset.pbtxt | 22 name: "output_shapes"
50 name: "output_shapes"
77 name: "output_shapes"
118 name: "output_shapes"
|
| D | ParallelMapDataset.pbtxt | 35 name: "output_shapes"
76 name: "output_shapes"
116 name: "output_shapes"
163 name: "output_shapes"
217 name: "output_shapes"
|
| D | MapDataset.pbtxt | 31 name: "output_shapes"
68 name: "output_shapes"
104 name: "output_shapes"
147 name: "output_shapes"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | PrefetchDataset.pbtxt | 22 name: "output_shapes"
50 name: "output_shapes"
77 name: "output_shapes"
111 name: "output_shapes"
|
| D | ParallelMapDataset.pbtxt | 35 name: "output_shapes"
76 name: "output_shapes"
116 name: "output_shapes"
163 name: "output_shapes"
217 name: "output_shapes"
|
| D | MapDataset.pbtxt | 31 name: "output_shapes"
68 name: "output_shapes"
104 name: "output_shapes"
147 name: "output_shapes"
|
| D | ModelDataset.pbtxt | 18 name: "output_shapes"
48 name: "output_shapes"
85 name: "output_shapes"
|
| D | RebatchDataset.pbtxt | 22 name: "output_shapes"
49 name: "output_shapes"
83 name: "output_shapes"
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | batch_matmul_test.py | 38 def BuildParams(self, graph_fn, dtype, input_shapes, output_shapes): argument
43 output_shapes=output_shapes,
45 output_mask=[[True] * len(s) for s in output_shapes],
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | test_util.py | 49 output_shapes = []
55 output_shapes.append(output_tensor.ShapeAsNumpy().tolist())
57 return output_shapes
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 120 TF_ShapeAndTypeList* output_shapes; in CheckOutputShapes() local
126 /*input_resource_shapes_and_types*/ nullptr, &output_shapes, in CheckOutputShapes()
129 CHECK_EQ(output_shapes->num_items, 1); in CheckOutputShapes()
131 int num_dims = output_shapes->items[0].num_dims; in CheckOutputShapes()
132 int64_t* dims = output_shapes->items[0].dims; in CheckOutputShapes()
145 TF_DeleteShapeAndTypeList(output_shapes); in CheckOutputShapes()
|