Home
last modified time | relevance | path

Searched refs:DimensionHandle (Results 1 – 25 of 57) sorted by relevance

123

/external/tensorflow/tensorflow/core/framework/
Dshape_inference.h56 class DimensionHandle {
58 DimensionHandle() {} in DimensionHandle() function
59 bool SameHandle(DimensionHandle d) const { return ptr_ == d.ptr_; } in SameHandle()
63 DimensionHandle(const Dimension* dim) { ptr_ = dim; } in DimensionHandle() function
86 Shape(const std::vector<DimensionHandle>& dims);
90 const std::vector<DimensionHandle> dims_;
127 DimensionOrConstant(DimensionHandle dim);
133 DimensionHandle dim;
342 DimensionHandle Dim(ShapeHandle s, int64 idx) { in Dim()
349 static DimensionHandle DimKnownRank(ShapeHandle s, int64 idx) { in DimKnownRank()
[all …]
Dcommon_shape_fns.cc140 shape_inference::DimensionHandle input_size, in GetWindowedOutputSizeFromDimsV2()
143 int64 padding_after, shape_inference::DimensionHandle* output_size) { in GetWindowedOutputSizeFromDimsV2()
162 DimensionHandle window_size; in GetWindowedOutputSizeFromDimsV2()
187 shape_inference::DimensionHandle input_size, in GetWindowedOutputSizeFromDims()
189 Padding padding_type, shape_inference::DimensionHandle* output_size) { in GetWindowedOutputSizeFromDims()
223 DimensionHandle output_rows = transpose_a ? c->Dim(a, 1) : c->Dim(a, 0); in MatMulShape()
224 DimensionHandle output_cols = transpose_b ? c->Dim(b, 0) : c->Dim(b, 1); in MatMulShape()
227 DimensionHandle inner_a = transpose_a ? c->Dim(a, 0) : c->Dim(a, 1); in MatMulShape()
228 DimensionHandle inner_b = transpose_b ? c->Dim(b, 1) : c->Dim(b, 0); in MatMulShape()
229 DimensionHandle merged; in MatMulShape()
[all …]
Dshape_inference.cc293 DimensionHandle dim = Dim(handle, i); in ShapeHandleToProto()
311 DimensionHandle InferenceContext::NumElements(ShapeHandle s) { in NumElements()
343 string InferenceContext::DebugString(DimensionHandle d) { in DebugString()
377 std::vector<DimensionHandle> dims; in WithRank()
421 Status InferenceContext::WithValue(DimensionHandle dim, int64 value, in WithValue()
422 DimensionHandle* out) { in WithValue()
429 DimensionHandle d = MakeDim(value); in WithValue()
437 void InferenceContext::Relax(DimensionHandle d_old, DimensionHandle d_new, in Relax()
438 DimensionHandle* out) { in Relax()
463 Status InferenceContext::Merge(DimensionHandle d0, DimensionHandle d1, in Merge()
[all …]
/external/tensorflow/tensorflow/core/ops/
Daudio_ops.cc25 using shape_inference::DimensionHandle;
33 DimensionHandle channels_dim; in DecodeWavShapeFn()
45 DimensionHandle samples_dim; in DecodeWavShapeFn()
78 DimensionHandle input_length = c->Dim(input, 0); in SpectrogramShapeFn()
79 DimensionHandle input_channels = c->Dim(input, 1); in SpectrogramShapeFn()
81 DimensionHandle output_length; in SpectrogramShapeFn()
96 DimensionHandle output_channels = in SpectrogramShapeFn()
113 DimensionHandle spectrogram_channels = c->Dim(spectrogram, 0); in MfccShapeFn()
114 DimensionHandle spectrogram_length = c->Dim(spectrogram, 1); in MfccShapeFn()
116 DimensionHandle output_channels = c->MakeDim(dct_coefficient_count); in MfccShapeFn()
Dimage_ops.cc22 using shape_inference::DimensionHandle;
30 Status SetOutputToSizedImage(InferenceContext* c, DimensionHandle batch_dim, in SetOutputToSizedImage()
31 int size_input_idx, DimensionHandle channel_dim) { in SetOutputToSizedImage()
35 DimensionHandle unused; in SetOutputToSizedImage()
40 DimensionHandle width; in SetOutputToSizedImage()
41 DimensionHandle height; in SetOutputToSizedImage()
72 DimensionHandle channels_dim; in DecodeImageShapeFn()
102 DimensionHandle last_dim; in ColorspaceShapeFn()
124 DimensionHandle unused; in NMSShapeFn()
150 DimensionHandle unused; in CombinedNMSShapeFn()
[all …]
Dlinalg_ops.cc22 using shape_inference::DimensionHandle;
34 DimensionHandle d; in MakeBatchSquareMatrix()
71 DimensionHandle m; in MatrixSolveShapeFn()
74 DimensionHandle n = c->Dim(lhs, -1); in MatrixSolveShapeFn()
93 DimensionHandle n; in SelfAdjointEigV2ShapeFn()
119 DimensionHandle n; in LuShapeFn()
144 DimensionHandle m = c->Dim(input, -2); in QrShapeFn()
145 DimensionHandle n = c->Dim(input, -1); in QrShapeFn()
146 DimensionHandle p; in QrShapeFn()
175 DimensionHandle m = c->Dim(input, -2); in SvdShapeFn()
[all …]
Dstring_ops.cc34 using shape_inference::DimensionHandle;
246 DimensionHandle pos_dim = c->Dim(pos_shape, i); in __anonb4ab91930702()
247 DimensionHandle len_dim = c->Dim(len_shape, i); in __anonb4ab91930702()
282 std::vector<DimensionHandle> dims(1); in __anonb4ab91930802()
309 DimensionHandle num_row_splits; in __anonb4ab91930902()
310 DimensionHandle input_size = c->NumElements(c->input(0)); in __anonb4ab91930902()
315 DimensionHandle num_chars = c->UnknownDim(); in __anonb4ab91930902()
331 DimensionHandle num_row_splits; in __anonb4ab91930a02()
332 DimensionHandle input_size = c->NumElements(c->input(0)); in __anonb4ab91930a02()
337 DimensionHandle num_chars = c->UnknownDim(); in __anonb4ab91930a02()
Darray_ops.cc27 using shape_inference::DimensionHandle;
60 std::vector<DimensionHandle> dims(num_dims); in PadKnown()
78 DimensionHandle unused; in PadShapeFn()
83 DimensionHandle n_dim = c->Dim(paddings, 0); in PadShapeFn()
118 DimensionHandle perm_elems = c->NumElements(perm_shape); in TransposeShapeFn()
144 std::vector<DimensionHandle> dims; in TransposeShapeFn()
197 DimensionHandle known_out_elems = c->NumElements(out); in SetOutputShapeForReshape()
201 DimensionHandle dim = c->Dim(out, i); in SetOutputShapeForReshape()
215 DimensionHandle known_in_elems = c->NumElements(in); in SetOutputShapeForReshape()
219 DimensionHandle dim = c->Dim(in, i); in SetOutputShapeForReshape()
[all …]
Dctc_ops.cc21 using shape_inference::DimensionHandle;
48 DimensionHandle unused; in __anon78c05cd60102()
54 DimensionHandle batch_size; in __anon78c05cd60102()
80 DimensionHandle batch_size; in __anon78c05cd60202()
84 DimensionHandle total_decoded_outputs = c->UnknownDim(); in __anon78c05cd60202()
110 DimensionHandle batch_size; in __anon78c05cd60302()
Dset_ops.cc22 using shape_inference::DimensionHandle;
53 DimensionHandle output_rank; in __anon967df2c80102()
113 DimensionHandle input1_rank_dim = c->Dim(input1_shape_shape, 0); in __anon967df2c80202()
115 DimensionHandle output_rank_dim; in __anon967df2c80202()
162 DimensionHandle input0_rank_dim = c->Dim(input0_shape_shape, 0); in __anon967df2c80302()
163 DimensionHandle input1_rank_dim = c->Dim(input1_shape_shape, 0); in __anon967df2c80302()
164 DimensionHandle output_rank_dim; in __anon967df2c80302()
Dlookup_ops.cc24 using shape_inference::DimensionHandle;
34 DimensionHandle unused_handle; in TwoElementVectorInputsAndScalarOutputs()
47 DimensionHandle unused_handle; in ScalarAndTwoElementVectorInputsAndScalarOutputs()
80 DimensionHandle unused_dim; in __anon8a0811960202()
134 DimensionHandle dim = c->Dim(key_shape_and_type.shape, d); in ValidateTableResourceHandle()
138 std::vector<DimensionHandle> keys_prefix_vec; in ValidateTableResourceHandle()
196 DimensionHandle unused_dim; in __anon8a0811960402()
251 DimensionHandle unused_dim; in __anon8a0811960702()
293 DimensionHandle unused_dim; in __anon8a0811960902()
451 DimensionHandle unused_dim; in __anon8a0811960e02()
[all …]
Dstateless_random_ops.cc21 using shape_inference::DimensionHandle;
29 DimensionHandle unused; in StatelessShape()
83 DimensionHandle unused_dim; in __anon530aab340202()
88 DimensionHandle num_samples; in __anon530aab340202()
Dsparse_ops.cc22 using shape_inference::DimensionHandle;
96 DimensionHandle unused_dim; in __anona7ffb82a0402()
111 DimensionHandle output_right = c->Dim(b, adjoint_b ? 0 : 1); in __anona7ffb82a0402()
112 DimensionHandle output_left = c->Dim(a_shape, adjoint_a ? 1 : 0); in __anona7ffb82a0402()
113 DimensionHandle inner_left = c->Dim(a_shape, adjoint_a ? 0 : 1); in __anona7ffb82a0402()
114 DimensionHandle inner_right = c->Dim(b, adjoint_b ? 1 : 0); in __anona7ffb82a0402()
161 DimensionHandle unused; in __anona7ffb82a0702()
180 DimensionHandle unused; in __anona7ffb82a0802()
219 DimensionHandle output_row_count = c->MakeDim(0ll); in __anona7ffb82a0a02()
222 DimensionHandle output_ind_cols = c->UnknownDim(); in __anona7ffb82a0a02()
[all …]
Ddata_flow_ops.cc23 using shape_inference::DimensionHandle;
163 DimensionHandle unused_handle; in TwoElementVectorInputsAndScalarOutputs()
601 DimensionHandle unused_dim; in __anon4a7ac08d1202()
623 DimensionHandle unused_dim; in __anon4a7ac08d1302()
639 std::vector<DimensionHandle> dims; in __anon4a7ac08d1302()
666 DimensionHandle unused_dim; in __anon4a7ac08d1402()
693 DimensionHandle unused_dim; in __anon4a7ac08d1502()
719 DimensionHandle unused_dim; in __anon4a7ac08d1602()
754 DimensionHandle unused_dim; in __anon4a7ac08d1702()
783 DimensionHandle unused_dim; in __anon4a7ac08d1802()
[all …]
Dnn_ops.cc26 using shape_inference::DimensionHandle;
42 std::vector<DimensionHandle> output_dims; in FractionalPoolShapeFn()
44 DimensionHandle d = c->Dim(input, i); in FractionalPoolShapeFn()
113 DimensionHandle last_dim = c->Dim(input, 3); in __anon3e672dd80302()
147 DimensionHandle last_dim = c->Dim(input, 3); in __anon3e672dd80402()
360 DimensionHandle new_height = c->UnknownDim(); in CommonFusedConvCalculations()
361 DimensionHandle new_width = c->UnknownDim(); in CommonFusedConvCalculations()
380 std::vector<DimensionHandle> output_dims; in CommonFusedConvCalculations()
382 DimensionHandle dim = c->Dim(resized, i); in CommonFusedConvCalculations()
411 DimensionHandle batch_size_dim = c->Dim(padded, 0); in CommonFusedConvCalculations()
[all …]
Dragged_conversion_ops.cc21 using shape_inference::DimensionHandle;
61 DimensionHandle dense_dims = in RaggedTensorToSparseShapeFn()
65 DimensionHandle num_values = c->NumElements(rt_dense_values); in RaggedTensorToSparseShapeFn()
Dcandidate_sampling_ops.cc21 using shape_inference::DimensionHandle;
35 DimensionHandle batch_size = c->Dim(true_classes_shape, 0); in CandidateSamplerShapeFn()
151 DimensionHandle unused; in __anon78463f730202()
/external/tensorflow/tensorflow/contrib/rnn/ops/
Dlstm_ops.cc21 using shape_inference::DimensionHandle;
50 DimensionHandle batch_size = c->Dim(x, 0); in __anon4050c6010102()
51 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anon4050c6010102()
137 DimensionHandle batch_size = c->Dim(x, 0); in __anon4050c6010202()
138 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anon4050c6010202()
139 DimensionHandle cell_size_times_4; in __anon4050c6010202()
205 DimensionHandle timelen = c->Dim(x, 0); in __anon4050c6010302()
206 DimensionHandle batch_size = c->Dim(x, 1); in __anon4050c6010302()
207 DimensionHandle cell_size; in __anon4050c6010302()
Dgru_ops.cc19 using tensorflow::shape_inference::DimensionHandle;
40 DimensionHandle batch_size = c->Dim(x, 0); in __anon07a5ee8f0102()
41 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anon07a5ee8f0102()
119 DimensionHandle batch_size = c->Dim(x, 0); in __anon07a5ee8f0202()
120 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anon07a5ee8f0202()
121 DimensionHandle twice_cell_size = c->Dim(w_ru, 1); in __anon07a5ee8f0202()
/external/tensorflow/tensorflow/contrib/image/ops/
Dimage_ops.cc22 using shape_inference::DimensionHandle;
30 Status SetOutputToSizedImage(InferenceContext* c, DimensionHandle batch_dim, in SetOutputToSizedImage()
31 int size_input_idx, DimensionHandle channel_dim) { in SetOutputToSizedImage()
35 DimensionHandle unused; in SetOutputToSizedImage()
40 DimensionHandle width; in SetOutputToSizedImage()
41 DimensionHandle height; in SetOutputToSizedImage()
Dsingle_image_random_dot_stereograms_ops.cc22 using shape_inference::DimensionHandle;
55 DimensionHandle x_dim = c->Dim(output_image_shape, 0); in __anonbda156eb0102()
56 DimensionHandle y_dim = c->Dim(output_image_shape, 1); in __anonbda156eb0102()
/external/tensorflow/tensorflow/contrib/reduce_slice_ops/ops/
Dreduce_slice_ops.cc21 using shape_inference::DimensionHandle;
29 DimensionHandle dimhandle; in ReduceSliceShapeFn()
30 DimensionHandle dim_axis = c->UnknownDim(); in ReduceSliceShapeFn()
43 DimensionHandle raw_dim_axis; in ReduceSliceShapeFn()
/external/tensorflow/tensorflow/contrib/fused_conv/ops/
Dfused_conv2d_bias_activation_op.cc51 using shape_inference::DimensionHandle; in __anon4e6e467d0102()
69 DimensionHandle output_depth_dim = in __anon4e6e467d0102()
77 DimensionHandle bias_dim = c->Dim(bias_shape, 0); in __anon4e6e467d0102()
/external/tensorflow/tensorflow/contrib/boosted_trees/ops/
Dsplit_handler_ops.cc21 using shape_inference::DimensionHandle;
44 DimensionHandle unused_dim; in __anon295dc3ce0102()
116 DimensionHandle unused_dim; in __anon295dc3ce0202()
187 DimensionHandle unused_dim; in __anon295dc3ce0302()
/external/tensorflow/tensorflow/contrib/tensor_forest/ops/
Dmodel_ops.cc21 using shape_inference::DimensionHandle;
95 DimensionHandle num_points = c->UnknownDim(); in __anon5b1697710102()
130 DimensionHandle num_points = c->UnknownDim(); in __anon5b1697710202()

123