| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 94 class ShapeHandle {
96 ShapeHandle() {} in ShapeHandle() function
97 bool SameHandle(ShapeHandle s) const { return ptr_ == s.ptr_; } in SameHandle()
101 ShapeHandle(const Shape* shape) { ptr_ = shape; } in ShapeHandle() function
135 ShapeAndType(ShapeHandle s, DataType t) : shape(s), dtype(t) {} in ShapeAndType()
136 ShapeAndType(ShapeHandle s, DataType t, SpecializedType specialized_t) in ShapeAndType()
139 ShapeHandle shape;
171 const std::vector<ShapeHandle>& input_shapes,
173 const std::vector<ShapeHandle>& input_tensors_as_shapes,
219 bool MergeInput(int idx, ShapeHandle shape) { in MergeInput()
[all …]
|
| D | common_shape_fns.h | 52 ShapeHandle out; in UnchangedShapeWithRank()
61 ShapeHandle out; in UnchangedShapeWithRankAtLeast()
70 ShapeHandle out; in UnchangedShapeWithRankAtMost()
89 ShapeHandle out; in MergeBothInputsShapeFn()
103 DimensionOrConstant C, ShapeHandle* out,
221 ShapeHandle shape_x,
222 ShapeHandle shape_y,
224 ShapeHandle* out);
230 ShapeHandle out; in BroadcastBinaryOpOutputShapeFn()
251 Status ValidateSparseTensor(InferenceContext* c, ShapeHandle indices_shape,
[all …]
|
| D | shape_inference.cc | 42 std::vector<ShapeHandle> input_tensors_as_shape_handles; in InferenceContext()
45 ShapeHandle shape; in InferenceContext()
56 ShapeHandle shape; in InferenceContext()
87 const std::vector<ShapeHandle>& input_shapes, in InferenceContext()
89 const std::vector<ShapeHandle>& input_tensors_as_shapes, in InferenceContext()
119 const std::vector<ShapeHandle>& shapes) { in set_output()
139 std::vector<ShapeHandle>* output) const { in input()
153 std::vector<ShapeHandle>* output) const { in output()
168 const std::vector<ShapeHandle>& input_tensors_as_shapes) { in PreInputInit()
228 void InferenceContext::ShapeHandleToProto(ShapeHandle handle, in ShapeHandleToProto()
[all …]
|
| D | common_shape_fns.cc | 111 ShapeHandle a; in MatMulShape()
114 ShapeHandle b; in MatMulShape()
180 gtl::InlinedVector<ShapeHandle, 2> input_bcast_shapes(c->num_inputs()); in EinsumShape()
184 ShapeHandle input_shape = c->input(i); in EinsumShape()
254 ShapeHandle output_bcast_shape; in EinsumShape()
310 ShapeHandle a_shape; in BatchMatMulV2Shape()
311 ShapeHandle b_shape; in BatchMatMulV2Shape()
329 ShapeHandle a_batch_shape; in BatchMatMulV2Shape()
330 ShapeHandle b_batch_shape; in BatchMatMulV2Shape()
331 ShapeHandle output_batch_shape; in BatchMatMulV2Shape()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | linalg_ops.cc | 24 using shape_inference::ShapeHandle;
29 Status MakeBatchSquareMatrix(InferenceContext* c, ShapeHandle input, in MakeBatchSquareMatrix()
30 ShapeHandle* out) { in MakeBatchSquareMatrix()
31 ShapeHandle s; in MakeBatchSquareMatrix()
37 ShapeHandle batch_shape; in MakeBatchSquareMatrix()
44 ShapeHandle out; in BatchUnchangedSquareShapeFn()
52 ShapeHandle lhs; in BandedTriangularSolveShapeFn()
53 ShapeHandle rhs; in BandedTriangularSolveShapeFn()
72 ShapeHandle lhs_batch_shape; in BandedTriangularSolveShapeFn()
73 ShapeHandle rhs_batch_shape; in BandedTriangularSolveShapeFn()
[all …]
|
| D | sparse_csr_matrix_ops.cc | 27 using shape_inference::ShapeHandle;
31 ShapeHandle variant; in GetVariantInput()
45 const ShapeHandle& matrix_shape, in ValidateSquareMatrixShape()
47 ShapeHandle out; in ValidateSquareMatrixShape()
69 ShapeHandle dense_shape; in __anon59a9d6c30102()
96 ShapeHandle sparse_matrix = sparse_matrix_shape_and_type.shape; in __anon59a9d6c30202()
102 ShapeHandle indices = c->Matrix(c->UnknownDim(), rank); in __anon59a9d6c30202()
103 ShapeHandle values = c->Vector(c->UnknownDim()); in __anon59a9d6c30202()
104 ShapeHandle dense_shape = c->Vector(rank); in __anon59a9d6c30202()
117 ShapeHandle dense_shape = c->input(0); in __anon59a9d6c30302()
[all …]
|
| D | sparse_ops.cc | 24 using shape_inference::ShapeHandle;
29 ShapeHandle unused; in SparseSparseMinOrMaxShapeFn()
53 ShapeHandle a_indices; in __anon36a8fc880202()
54 ShapeHandle b_indices; in __anon36a8fc880202()
76 ShapeHandle a_shape; in __anon36a8fc880302()
97 ShapeHandle unused; in __anon36a8fc880402()
98 ShapeHandle b; in __anon36a8fc880402()
99 ShapeHandle a_shape; in __anon36a8fc880402()
128 ShapeHandle unused; in __anon36a8fc880502()
144 ShapeHandle unused; in __anon36a8fc880602()
[all …]
|
| D | stateful_random_ops.cc | 24 using shape_inference::ShapeHandle; in StatefulRandomShape()
26 ShapeHandle unused; in StatefulRandomShape()
29 ShapeHandle out; in StatefulRandomShape()
60 using shape_inference::ShapeHandle; in __anon0d38c5e20102()
62 ShapeHandle unused; in __anon0d38c5e20102()
77 ShapeHandle out; in __anon0d38c5e20102()
88 shape_inference::ShapeHandle unused; in __anon0d38c5e20202()
100 shape_inference::ShapeHandle unused; in __anon0d38c5e20302()
114 using shape_inference::ShapeHandle; in __anon0d38c5e20402()
115 ShapeHandle out; in __anon0d38c5e20402()
[all …]
|
| D | image_ops.cc | 24 using shape_inference::ShapeHandle;
33 ShapeHandle size; in SetOutputToSizedImage()
63 ShapeHandle input; in ResizeShapeFn()
70 ShapeHandle unused; in DecodeImageShapeFn()
91 ShapeHandle unused; in DecodeImageV2ShapeFn()
125 ShapeHandle unused; in EncodeImageShapeFn()
132 ShapeHandle input; in ColorspaceShapeFn()
138 ShapeHandle out; in ColorspaceShapeFn()
147 ShapeHandle boxes; in NMSShapeFn()
149 ShapeHandle scores; in NMSShapeFn()
[all …]
|
| D | boosted_trees_ops.cc | 29 using shape_inference::ShapeHandle;
37 shape_inference::ShapeHandle unused_input; in __anon770f674c0102()
63 shape_inference::ShapeHandle node_id_range_shape; in __anon770f674c0202()
64 shape_inference::ShapeHandle unused_shape; in __anon770f674c0202()
69 shape_inference::ShapeHandle summary_shape_base; in __anon770f674c0202()
75 shape_inference::ShapeHandle summary_shape; in __anon770f674c0202()
87 std::vector<shape_inference::ShapeHandle> output_shapes_vec( in __anon770f674c0202()
92 std::vector<shape_inference::ShapeHandle> output_shapes_contribs( in __anon770f674c0202()
118 shape_inference::ShapeHandle node_id_range_shape; in __anon770f674c0302()
119 shape_inference::ShapeHandle unused_shape; in __anon770f674c0302()
[all …]
|
| D | list_ops.cc | 52 shape_inference::ShapeHandle GetElementShapeFromHandleData( in GetElementShapeFromHandleData()
67 shape_inference::ShapeHandle element_shape; in __anonabedecb60202()
85 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anonabedecb60302()
102 shape_inference::ShapeHandle ignored; in __anonabedecb60302()
119 shape_inference::ShapeHandle input_handles; in __anonabedecb60402()
122 shape_inference::ShapeHandle tensor; in __anonabedecb60402()
132 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anonabedecb60402()
149 shape_inference::ShapeHandle ignored; in __anonabedecb60402()
174 shape_inference::ShapeHandle tensor_shape = c->UnknownShape(); in __anonabedecb60502()
191 shape_inference::ShapeHandle ignored; in __anonabedecb60502()
[all …]
|
| D | data_flow_ops.cc | 25 using shape_inference::ShapeHandle;
29 Status DequeueManyV2Shape(InferenceContext* c, ShapeHandle n_shape) { in DequeueManyV2Shape()
33 ShapeHandle combined_shape; in DequeueManyV2Shape()
58 ShapeHandle data_shape = c->input(0); in __anonea01102b0202()
59 ShapeHandle partitions_shape = c->input(1); in __anonea01102b0202()
68 ShapeHandle unused; in __anonea01102b0202()
74 ShapeHandle unknown_dim0 = c->MakeShape({c->UnknownDim()}); in __anonea01102b0202()
76 ShapeHandle data_suffix_shape; in __anonea01102b0202()
78 ShapeHandle result_shape; in __anonea01102b0202()
97 ShapeHandle extra_shape = c->UnknownShape(); in DynamicStitchShapeFunction()
[all …]
|
| D | lookup_ops.cc | 27 using shape_inference::ShapeHandle;
33 ShapeHandle handle; in TwoElementVectorInputsAndScalarOutputs()
46 ShapeHandle handle; in ScalarAndTwoElementVectorInputsAndScalarOutputs()
78 ShapeHandle handle; in __anon18b155f40202()
84 ShapeHandle unused; in __anon18b155f40202()
116 Status ValidateTableResourceHandle(InferenceContext* c, ShapeHandle keys, in ValidateTableResourceHandle()
151 ShapeHandle keys_prefix = c->MakeShape(keys_prefix_vec); in ValidateTableResourceHandle()
169 ShapeHandle handle; in __anon18b155f40302()
192 ShapeHandle handle; in __anon18b155f40402()
208 ShapeHandle handle; in __anon18b155f40502()
[all …]
|
| D | ctc_ops.cc | 23 using shape_inference::ShapeHandle;
39 ShapeHandle inputs; in __anonbca2cdf40102()
40 ShapeHandle labels_indices; in __anonbca2cdf40102()
41 ShapeHandle labels_values; in __anonbca2cdf40102()
42 ShapeHandle sequence_length; in __anonbca2cdf40102()
76 ShapeHandle inputs; in __anonbca2cdf40202()
77 ShapeHandle labels_indices; in __anonbca2cdf40202()
78 ShapeHandle labels_values; in __anonbca2cdf40202()
79 ShapeHandle sequence_length; in __anonbca2cdf40202()
112 ShapeHandle inputs; in __anonbca2cdf40302()
[all …]
|
| D | stateless_random_ops.cc | 23 using shape_inference::ShapeHandle;
27 ShapeHandle seed; in StatelessShape()
33 ShapeHandle out; in StatelessShape()
65 ShapeHandle unused; in __anona1237d120102()
100 ShapeHandle seed; in __anona1237d120202()
105 ShapeHandle logits_shape; in __anona1237d120202()
106 ShapeHandle unused; in __anona1237d120202()
140 ShapeHandle seed; in __anona1237d120302()
145 ShapeHandle bcast_means_stddevs; in __anona1237d120302()
146 ShapeHandle bcast_except_maxvals; in __anona1237d120302()
[all …]
|
| D | io_ops.cc | 25 using shape_inference::ShapeHandle;
30 ShapeHandle unused; in ScalarInputsAndOutputs()
41 ShapeHandle handle; in TwoElementVectorAndScalarOutputs()
68 ShapeHandle unused; in __anon52e493120202()
69 ShapeHandle s; in __anon52e493120202()
94 ShapeHandle shape0, shape1, shape2; in __anon52e493120302()
121 ShapeHandle shape_handle; in __anon52e493120302()
138 ShapeHandle unused; in __anon52e493120402()
151 ShapeHandle unused; in __anon52e493120502()
152 ShapeHandle s; in __anon52e493120502()
[all …]
|
| D | ragged_array_ops.cc | 23 using shape_inference::ShapeHandle;
84 ShapeHandle out_values = c->UnknownShapeOfRank(1); in __anon9deb1dc20102()
85 ShapeHandle out_splits = c->UnknownShapeOfRank(1); in __anon9deb1dc20102()
91 ShapeHandle row_splits = c->input(i + ragged_splits_start); in __anon9deb1dc20102()
101 ShapeHandle dense_input = c->input(i + dense_start); in __anon9deb1dc20102()
104 ShapeHandle row_splits = c->Vector(batch_size + 1); in __anon9deb1dc20102()
129 ShapeHandle indices = c->input(PARAMS_RAGGED_RANK + 1); in RaggedGatherShapeFn()
135 ShapeHandle splits = c->input(i); in RaggedGatherShapeFn()
140 ShapeHandle params_dense_values = c->input(PARAMS_RAGGED_RANK); in RaggedGatherShapeFn()
150 ShapeHandle value = c->UnknownShape(); in RaggedGatherShapeFn()
[all …]
|
| D | array_ops.cc | 36 using shape_inference::ShapeHandle;
64 Status PadKnown(InferenceContext* c, ShapeHandle input, in PadKnown()
83 ShapeHandle paddings; in PadShapeFn()
89 ShapeHandle input = c->input(0); in PadShapeFn()
122 ShapeHandle input = c->input(0); in TransposeShapeFn()
123 ShapeHandle perm_shape = c->input(1); in TransposeShapeFn()
188 ShapeHandle in = c->input(0); in SetOutputShapeForReshape()
189 ShapeHandle out; in SetOutputShapeForReshape()
294 ShapeHandle passed_shape; in __anon42d741190202()
300 ShapeHandle cur; in __anon42d741190202()
[all …]
|
| D | parsing_ops.cc | 26 using shape_inference::ShapeHandle;
33 const ShapeHandle& prefix, InferenceContext* c, in AddDenseOutputShapes()
36 ShapeHandle s; in AddDenseOutputShapes()
45 void AddSparseOutputShapes(int num_sparse, const ShapeHandle input_shape, in AddSparseOutputShapes()
88 void AddDenseLengthsShapes(int num_dense, const ShapeHandle& shape, in AddDenseLengthsShapes()
107 ShapeHandle out; in __anonf8f6e66e0202()
132 ShapeHandle out; in __anonf8f6e66e0302()
164 ShapeHandle input; in __anonf8f6e66e0402()
166 ShapeHandle names; in __anonf8f6e66e0402()
208 ShapeHandle input; in __anonf8f6e66e0502()
[all …]
|
| D | math_ops.cc | 25 using shape_inference::ShapeHandle;
35 ShapeHandle cur = c->input(c->num_inputs() - 1); in __anon725c8f640102()
654 ShapeHandle output = c->UnknownShape(); in __anon725c8f640202()
656 ShapeHandle some_non_scalar; in __anon725c8f640202()
658 ShapeHandle in = c->input(i); in __anon725c8f640202()
716 ShapeHandle x = c->input(0); \
717 ShapeHandle y = c->input(1); \
718 ShapeHandle output; \
743 ShapeHandle data_x = c->input(0); in __anon725c8f640302()
744 ShapeHandle data_y = c->input(1); in __anon725c8f640302()
[all …]
|
| D | set_ops.cc | 24 using shape_inference::ShapeHandle;
54 ShapeHandle input0_shape = c->input(0); in __anonda6063e60102()
58 ShapeHandle input1_shape = c->input(1); in __anonda6063e60102()
64 ShapeHandle group0_shape; in __anonda6063e60102()
67 ShapeHandle group1_shape; in __anonda6063e60102()
70 ShapeHandle unused_shape; in __anonda6063e60102()
76 ShapeHandle input1_shape = c->input(1); in __anonda6063e60102()
109 ShapeHandle input1_shape_shape = c->input(3); in __anonda6063e60202()
116 ShapeHandle input0_shape = c->input(0); in __anonda6063e60202()
155 ShapeHandle input0_shape_shape = c->input(2); in __anonda6063e60302()
[all …]
|
| D | training_ops.cc | 23 using shape_inference::ShapeHandle;
26 ShapeHandle ShapeOrHandleShape(InferenceContext* c, int input) { in ShapeOrHandleShape()
36 ShapeHandle ShapeOrHandleShape<true>(InferenceContext* c, int input) { in ShapeOrHandleShape()
53 ShapeHandle* s) { in HandleGradAndIndicesInputs()
54 ShapeHandle grad = ShapeOrHandleShape<is_resource>(c, grad_idx); in HandleGradAndIndicesInputs()
60 ShapeHandle indices; in HandleGradAndIndicesInputs()
65 ShapeHandle grad_unknown_first; in HandleGradAndIndicesInputs()
75 ShapeHandle unused; in ApplyGradientDescentShapeFn()
76 ShapeHandle s = ShapeOrHandleShape<is_resource>(c, 0); // var in ApplyGradientDescentShapeFn()
104 ShapeHandle unused; in ApplyProximalGradientDescentShapeFn()
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/ops/ |
| D | tpu_partitioned_input_op.cc | 24 using shape_inference::ShapeHandle;
26 ShapeHandle _UpdatePartitionDim(InferenceContext* c, const ShapeHandle handle, in _UpdatePartitionDim()
28 ShapeHandle newoutput0; in _UpdatePartitionDim()
48 ShapeHandle cur = c->input(c->num_inputs() - 1); in __anon2b8a3a590102()
57 ShapeHandle newoutput0 = _UpdatePartitionDim(c, cur, partition_dim); in __anon2b8a3a590102()
63 ShapeHandle previous_shape_handle; in __anon2b8a3a590102()
69 ShapeHandle shape_handle = shapes_and_types->at(0).shape; in __anon2b8a3a590102()
76 ShapeHandle tmp; in __anon2b8a3a590102()
90 ShapeHandle newoutput0 = in __anon2b8a3a590102()
|
| D | topk_ops.cc | 22 using shape_inference::ShapeHandle;
29 ShapeHandle input; in __anonee73ab400102()
32 ShapeHandle s; in __anonee73ab400102()
44 ShapeHandle input; in __anonee73ab400202()
50 ShapeHandle s; in __anonee73ab400202()
61 ShapeHandle input; in __anonee73ab400302()
73 ShapeHandle input; in __anonee73ab400402()
79 ShapeHandle s; in __anonee73ab400402()
|
| /external/tensorflow/tensorflow/c/ |
| D | ops.cc | 33 using ::tensorflow::shape_inference::ShapeHandle;
104 return reinterpret_cast<TF_ShapeHandle*>(new ShapeHandle); in TF_NewShapeHandle()
108 auto* handle = new ShapeHandle; in TF_ShapeInferenceContextScalar()
115 auto* handle = new ShapeHandle; in TF_ShapeInferenceContextVectorFromSize()
126 Status s = cc_ctx->Concatenate(*reinterpret_cast<ShapeHandle*>(first), in TF_ShapeInferenceContextConcatenateShapes()
127 *reinterpret_cast<ShapeHandle*>(second), in TF_ShapeInferenceContextConcatenateShapes()
128 reinterpret_cast<ShapeHandle*>(result)); in TF_ShapeInferenceContextConcatenateShapes()
150 auto* cc_result = reinterpret_cast<ShapeHandle*>(handle); in TF_ShapeInferenceContextGetInput()
158 return cc_ctx->RankKnown(*reinterpret_cast<ShapeHandle*>(handle)); in TF_ShapeInferenceContextRankKnown()
170 cc_ctx->set_output(i, *(reinterpret_cast<ShapeHandle*>(handle))); in TF_ShapeInferenceContextSetOutput()
[all …]
|