| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 99 class ShapeHandle {
101 ShapeHandle() {} in ShapeHandle() function
102 bool SameHandle(ShapeHandle s) const { return ptr_ == s.ptr_; } in SameHandle()
106 ShapeHandle(const Shape* shape) { ptr_ = shape; } in ShapeHandle() function
142 ShapeAndType(ShapeHandle s, DataType t) : shape(s), dtype(t) {} in ShapeAndType()
144 ShapeHandle shape;
175 const std::vector<ShapeHandle>& input_shapes,
177 const std::vector<ShapeHandle>& input_tensors_as_shapes,
246 bool MergeInput(int idx, ShapeHandle shape) { in MergeInput()
247 ShapeHandle new_shape; in MergeInput()
[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()
97 std::vector<ShapeHandle> input_tensors_as_shape_handles; in InferenceContext()
100 ShapeHandle shape; in InferenceContext()
111 ShapeHandle shape; in InferenceContext()
142 const std::vector<ShapeHandle>& input_shapes, in InferenceContext()
144 const std::vector<ShapeHandle>& input_tensors_as_shapes, in InferenceContext()
176 const std::vector<ShapeHandle>& shapes) { in set_output()
195 std::vector<ShapeHandle>* output) const { in input()
[all …]
|
| D | common_shape_fns.cc | 214 ShapeHandle a; in MatMulShape()
217 ShapeHandle b; in MatMulShape()
237 ShapeHandle input_shape; in BiasAddShape()
249 ShapeHandle bias_shape; in BiasAddShape()
261 ShapeHandle output_shape; in BiasAddShape()
264 ShapeHandle first; in BiasAddShape()
267 ShapeHandle last; in BiasAddShape()
273 ShapeHandle merged_bias = c->Vector(merged_bias_dim); in BiasAddShape()
275 ShapeHandle temp; in BiasAddShape()
279 ShapeHandle all_but_bias; in BiasAddShape()
[all …]
|
| D | common_shape_fns.h | 175 ShapeHandle out; in UnchangedShapeWithRank()
184 ShapeHandle out; in UnchangedShapeWithRankAtLeast()
193 ShapeHandle out; in UnchangedShapeWithRankAtMost()
212 ShapeHandle out; in MergeBothInputsShapeFn()
223 DimensionOrConstant C, ShapeHandle* out,
288 ShapeHandle shape_x,
289 ShapeHandle shape_y,
290 ShapeHandle* out);
296 ShapeHandle out; in BroadcastBinaryOpOutputShapeFn()
317 Status ValidateSparseTensor(InferenceContext* c, ShapeHandle indices_shape,
[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()
53 ShapeHandle lhs; in MatrixSolveShapeFn()
54 ShapeHandle rhs; in MatrixSolveShapeFn()
62 ShapeHandle lhs_batch_shape; in MatrixSolveShapeFn()
63 ShapeHandle rhs_batch_shape; in MatrixSolveShapeFn()
[all …]
|
| D | sparse_ops.cc | 24 using shape_inference::ShapeHandle;
29 ShapeHandle unused; in SparseSparseMinOrMaxShapeFn()
53 ShapeHandle a_indices; in __anone6e195410202()
54 ShapeHandle b_indices; in __anone6e195410202()
76 ShapeHandle a_shape; in __anone6e195410302()
97 ShapeHandle unused; in __anone6e195410402()
98 ShapeHandle b; in __anone6e195410402()
99 ShapeHandle a_shape; in __anone6e195410402()
128 ShapeHandle unused; in __anone6e195410502()
144 ShapeHandle unused; in __anone6e195410602()
[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 EncodeImageShapeFn()
98 ShapeHandle input; in ColorspaceShapeFn()
104 ShapeHandle out; in ColorspaceShapeFn()
113 ShapeHandle boxes; in NMSShapeFn()
115 ShapeHandle scores; in NMSShapeFn()
117 ShapeHandle max_output_size; in NMSShapeFn()
[all …]
|
| D | boosted_trees_ops.cc | 28 using shape_inference::ShapeHandle;
36 shape_inference::ShapeHandle unused_input; in __anon8a3fd5e50102()
62 shape_inference::ShapeHandle node_id_range_shape; in __anon8a3fd5e50202()
63 shape_inference::ShapeHandle unused_shape; in __anon8a3fd5e50202()
68 shape_inference::ShapeHandle summary_shape_base; in __anon8a3fd5e50202()
74 shape_inference::ShapeHandle summary_shape; in __anon8a3fd5e50202()
86 std::vector<shape_inference::ShapeHandle> output_shapes_vec( in __anon8a3fd5e50202()
91 std::vector<shape_inference::ShapeHandle> output_shapes_contribs( in __anon8a3fd5e50202()
105 shape_inference::ShapeHandle unused_input; in __anon8a3fd5e50302()
117 shape_inference::ShapeHandle unused_input; in __anon8a3fd5e50402()
[all …]
|
| D | lookup_ops.cc | 27 using shape_inference::ShapeHandle;
33 ShapeHandle handle; in TwoElementVectorInputsAndScalarOutputs()
46 ShapeHandle handle; in ScalarAndTwoElementVectorInputsAndScalarOutputs()
78 ShapeHandle handle; in __anonc8e9eead0202()
84 ShapeHandle unused; in __anonc8e9eead0202()
90 Status ValidateTableResourceHandle(InferenceContext* c, ShapeHandle keys, in ValidateTableResourceHandle()
143 ShapeHandle keys_prefix = c->MakeShape(keys_prefix_vec); in ValidateTableResourceHandle()
166 ShapeHandle handle; in __anonc8e9eead0302()
170 ShapeHandle keys; in __anonc8e9eead0302()
194 ShapeHandle handle; in __anonc8e9eead0402()
[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 __anona2f81a440202()
59 ShapeHandle partitions_shape = c->input(1); in __anona2f81a440202()
68 ShapeHandle unused; in __anona2f81a440202()
74 ShapeHandle unknown_dim0 = c->MakeShape({c->UnknownDim()}); in __anona2f81a440202()
76 ShapeHandle data_suffix_shape; in __anona2f81a440202()
78 ShapeHandle result_shape; in __anona2f81a440202()
97 ShapeHandle extra_shape = c->UnknownShape(); in DynamicStitchShapeFunction()
[all …]
|
| D | list_ops.cc | 46 shape_inference::ShapeHandle GetElementShapeFromHandleData( in GetElementShapeFromHandleData()
61 shape_inference::ShapeHandle element_shape; in __anon9187032f0202()
79 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anon9187032f0302()
96 shape_inference::ShapeHandle ignored; in __anon9187032f0302()
113 shape_inference::ShapeHandle input_handles; in __anon9187032f0402()
116 shape_inference::ShapeHandle tensor; in __anon9187032f0402()
126 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anon9187032f0402()
143 shape_inference::ShapeHandle ignored; in __anon9187032f0402()
168 shape_inference::ShapeHandle tensor_shape = c->UnknownShape(); in __anon9187032f0502()
185 shape_inference::ShapeHandle ignored; in __anon9187032f0502()
[all …]
|
| D | array_ops.cc | 29 using shape_inference::ShapeHandle;
57 Status PadKnown(InferenceContext* c, ShapeHandle input, in PadKnown()
76 ShapeHandle paddings; in PadShapeFn()
82 ShapeHandle input = c->input(0); in PadShapeFn()
115 ShapeHandle input = c->input(0); in TransposeShapeFn()
116 ShapeHandle perm_shape = c->input(1); in TransposeShapeFn()
181 ShapeHandle in = c->input(0); in SetOutputShapeForReshape()
182 ShapeHandle out; in SetOutputShapeForReshape()
288 ShapeHandle passed_shape; in __anondb9326b20202()
294 ShapeHandle cur; in __anondb9326b20202()
[all …]
|
| D | stateful_random_ops.cc | 22 using shape_inference::ShapeHandle; in StatefulRandomShape()
24 ShapeHandle unused; in StatefulRandomShape()
27 ShapeHandle out; in StatefulRandomShape()
56 using shape_inference::ShapeHandle; in __anonae3f3ebb0102()
58 ShapeHandle unused; in __anonae3f3ebb0102()
63 ShapeHandle out; in __anonae3f3ebb0102()
76 using shape_inference::ShapeHandle; in __anonae3f3ebb0202()
77 ShapeHandle out; in __anonae3f3ebb0202()
93 using shape_inference::ShapeHandle; in __anonae3f3ebb0302()
95 ShapeHandle out; in __anonae3f3ebb0302()
|
| D | io_ops.cc | 25 using shape_inference::ShapeHandle;
30 ShapeHandle unused; in ScalarInputsAndOutputs()
41 ShapeHandle handle; in TwoElementVectorAndScalarOutputs()
68 ShapeHandle unused; in __anonec23b74b0202()
69 ShapeHandle s; in __anonec23b74b0202()
94 ShapeHandle shape0, shape1, shape2; in __anonec23b74b0302()
121 ShapeHandle shape_handle; in __anonec23b74b0302()
138 ShapeHandle unused; in __anonec23b74b0402()
151 ShapeHandle unused; in __anonec23b74b0502()
152 ShapeHandle s; in __anonec23b74b0502()
[all …]
|
| D | math_ops.cc | 25 using shape_inference::ShapeHandle;
35 ShapeHandle cur = c->input(c->num_inputs() - 1); in __anon57f5a5dd0102()
128 ShapeHandle a_shape; in __anon57f5a5dd0202()
129 ShapeHandle b_shape; in __anon57f5a5dd0202()
142 ShapeHandle a_batch_dims; in __anon57f5a5dd0202()
143 ShapeHandle b_batch_dims; in __anon57f5a5dd0202()
144 ShapeHandle batch_dims; in __anon57f5a5dd0202()
154 ShapeHandle out; in __anon57f5a5dd0202()
626 ShapeHandle output = c->UnknownShape(); in __anon57f5a5dd0302()
628 ShapeHandle some_non_scalar; in __anon57f5a5dd0302()
[all …]
|
| D | ctc_ops.cc | 23 using shape_inference::ShapeHandle;
38 ShapeHandle inputs; in __anon7dc6794d0102()
39 ShapeHandle labels_indices; in __anon7dc6794d0102()
40 ShapeHandle labels_values; in __anon7dc6794d0102()
41 ShapeHandle sequence_length; in __anon7dc6794d0102()
73 ShapeHandle inputs; in __anon7dc6794d0202()
74 ShapeHandle sequence_length; in __anon7dc6794d0202()
103 ShapeHandle inputs; in __anon7dc6794d0302()
104 ShapeHandle sequence_length; in __anon7dc6794d0302()
125 ShapeHandle shape_v = c->Vector(2); in __anon7dc6794d0302()
|
| D | nn_ops.cc | 28 using shape_inference::ShapeHandle;
33 ShapeHandle input; in FractionalPoolShapeFn()
89 ShapeHandle s; in __anon4650176f0202()
110 ShapeHandle input; in __anon4650176f0302()
115 ShapeHandle vec; in __anon4650176f0302()
120 ShapeHandle out; in __anon4650176f0302()
142 ShapeHandle input; in __anon4650176f0402()
149 ShapeHandle vec; in __anon4650176f0402()
154 ShapeHandle dx; in __anon4650176f0402()
158 ShapeHandle vector_shape = c->Vector(last_dim); in __anon4650176f0402()
[all …]
|
| D | training_ops.cc | 23 using shape_inference::ShapeHandle;
25 static ShapeHandle ShapeOrHandleShape(InferenceContext* c, int input) { in ShapeOrHandleShape()
38 int grad_idx, ShapeHandle* s) { in HandleGradAndIndicesInputs()
39 ShapeHandle grad = ShapeOrHandleShape(c, grad_idx); in HandleGradAndIndicesInputs()
45 ShapeHandle indices; in HandleGradAndIndicesInputs()
51 ShapeHandle grad_unknown_first; in HandleGradAndIndicesInputs()
60 ShapeHandle unused; in ApplyGradientDescentShapeFn()
61 ShapeHandle s = ShapeOrHandleShape(c, 0); // var in ApplyGradientDescentShapeFn()
89 ShapeHandle unused; in ApplyProximalGradientDescentShapeFn()
90 ShapeHandle s = ShapeOrHandleShape(c, 0); // var in ApplyProximalGradientDescentShapeFn()
[all …]
|
| D | resource_variable_ops.cc | 26 using ::tensorflow::shape_inference::ShapeHandle;
110 ShapeHandle s; in __anon4d581ac00202()
143 ShapeHandle s; in __anon4d581ac00302()
191 ShapeHandle value_shape = c->input(1); in CreateAssignShapeFn()
192 ShapeHandle unused; in CreateAssignShapeFn()
241 ShapeHandle var_shape = (*handle_data)[0].shape; in VariableShapeShapeFn()
267 ShapeHandle indices_shape = c->input(1); in __anon4d581ac00402()
269 ShapeHandle unused; in __anon4d581ac00402()
282 ShapeHandle params_subshape1; in __anon4d581ac00402()
286 ShapeHandle params_subshape2; in __anon4d581ac00402()
[all …]
|
| D | set_ops.cc | 24 using shape_inference::ShapeHandle;
54 ShapeHandle input0_shape = c->input(0); in __anon9b840f3f0102()
58 ShapeHandle input1_shape = c->input(1); in __anon9b840f3f0102()
64 ShapeHandle group0_shape; in __anon9b840f3f0102()
67 ShapeHandle group1_shape; in __anon9b840f3f0102()
70 ShapeHandle unused_shape; in __anon9b840f3f0102()
76 ShapeHandle input1_shape = c->input(1); in __anon9b840f3f0102()
109 ShapeHandle input1_shape_shape = c->input(3); in __anon9b840f3f0202()
116 ShapeHandle input0_shape = c->input(0); in __anon9b840f3f0202()
155 ShapeHandle input0_shape_shape = c->input(2); in __anon9b840f3f0302()
[all …]
|
| D | parsing_ops.cc | 24 using shape_inference::ShapeHandle;
36 ShapeHandle out; in __anonb04296470102()
68 ShapeHandle input; in __anonb04296470202()
70 ShapeHandle unused; in __anonb04296470202()
87 ShapeHandle dense; in __anonb04296470202()
113 ShapeHandle input; in __anonb04296470302()
130 ShapeHandle dense; in __anonb04296470302()
171 ShapeHandle input; in __anonb04296470402()
175 ShapeHandle unused; in __anonb04296470402()
194 ShapeHandle s; in __anonb04296470402()
[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 __anon62f9110b0102()
81 ShapeHandle seed; in __anon62f9110b0202()
86 ShapeHandle logits_shape; in __anon62f9110b0202()
87 ShapeHandle unused; in __anon62f9110b0202()
|
| D | dataset_ops.cc | 109 shape_inference::ShapeHandle count_shape; in __anonfa0360990102()
121 shape_inference::ShapeHandle count_shape; in __anonfa0360990202()
133 shape_inference::ShapeHandle count_shape; in __anonfa0360990302()
171 shape_inference::ShapeHandle unused; in __anonfa0360990402()
240 shape_inference::ShapeHandle unused; in __anonfa0360990502()
256 shape_inference::ShapeHandle unused; in __anonfa0360990602()
270 shape_inference::ShapeHandle unused; in __anonfa0360990702()
286 shape_inference::ShapeHandle unused; in __anonfa0360990802()
308 shape_inference::ShapeHandle unused; in __anonfa0360990902()
325 shape_inference::ShapeHandle unused; in __anonfa0360990a02()
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/ops/ |
| D | split_handler_ops.cc | 23 using shape_inference::ShapeHandle;
45 ShapeHandle unused_shape; in __anond11615e50102()
48 ShapeHandle partition_ids_shape; in __anond11615e50102()
50 ShapeHandle bucket_ids_shape; in __anond11615e50102()
52 ShapeHandle gradients_shape; in __anond11615e50102()
56 ShapeHandle hessians_shape; in __anond11615e50102()
60 ShapeHandle bucket_boundaries_shape; in __anond11615e50102()
117 ShapeHandle unused_shape; in __anond11615e50202()
120 ShapeHandle partition_ids_shape; in __anond11615e50202()
122 ShapeHandle bucket_ids_shape; in __anond11615e50202()
[all …]
|
| D | stats_accumulator_ops.cc | 24 using shape_inference::ShapeHandle;
40 ShapeHandle unused_input; in __anon72691d0a0102()
66 ShapeHandle unused_input; in __anon72691d0a0202()
71 ShapeHandle partition_ids_shape; in __anon72691d0a0202()
74 ShapeHandle feature_ids_shape; in __anon72691d0a0202()
77 ShapeHandle gradients_shape; in __anon72691d0a0202()
82 ShapeHandle hessians_shape; in __anon72691d0a0202()
114 ShapeHandle unused_input; in __anon72691d0a0302()
151 ShapeHandle unused_input; in __anon72691d0a0402()
159 ShapeHandle partition_ids_shape; in __anon72691d0a0402()
[all …]
|