Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/framework/
Dshape_inference.h54 class DimensionHandle {
56 DimensionHandle() {} in DimensionHandle() function
57 bool SameHandle(DimensionHandle d) const { return ptr_ == d.ptr_; } in SameHandle()
61 DimensionHandle(const Dimension* dim) { ptr_ = dim; } in DimensionHandle() function
82 Shape(const std::vector<DimensionHandle>& dims);
86 const std::vector<DimensionHandle> dims_;
120 DimensionOrConstant(DimensionHandle dim);
126 DimensionHandle dim;
329 DimensionHandle Dim(ShapeHandle s, int64 idx) { in Dim()
336 static DimensionHandle DimKnownRank(ShapeHandle s, int64 idx) { in DimKnownRank()
[all …]
Dcommon_shape_fns.cc37 shape_inference::DimensionHandle input_size, in GetWindowedOutputSizeFromDimsV2()
40 int64 padding_after, shape_inference::DimensionHandle* output_size) { in GetWindowedOutputSizeFromDimsV2()
59 DimensionHandle window_size; in GetWindowedOutputSizeFromDimsV2()
84 shape_inference::DimensionHandle input_size, in GetWindowedOutputSizeFromDims()
86 Padding padding_type, shape_inference::DimensionHandle* output_size) { in GetWindowedOutputSizeFromDims()
120 DimensionHandle output_rows = transpose_a ? c->Dim(a, 1) : c->Dim(a, 0); in MatMulShape()
121 DimensionHandle output_cols = transpose_b ? c->Dim(b, 0) : c->Dim(b, 1); in MatMulShape()
124 DimensionHandle inner_a = transpose_a ? c->Dim(a, 0) : c->Dim(a, 1); in MatMulShape()
125 DimensionHandle inner_b = transpose_b ? c->Dim(b, 1) : c->Dim(b, 0); in MatMulShape()
126 DimensionHandle merged; in MatMulShape()
[all …]
Dshape_inference.cc236 DimensionHandle dim = Dim(handle, i); in ShapeHandleToProto()
254 DimensionHandle InferenceContext::NumElements(ShapeHandle s) { in NumElements()
286 string InferenceContext::DebugString(DimensionHandle d) { in DebugString()
319 std::vector<DimensionHandle> dims; in WithRank()
363 Status InferenceContext::WithValue(DimensionHandle dim, int64 value, in WithValue()
364 DimensionHandle* out) { in WithValue()
371 DimensionHandle d = MakeDim(value); in WithValue()
379 void InferenceContext::Relax(DimensionHandle d_old, DimensionHandle d_new, in Relax()
380 DimensionHandle* out) { in Relax()
405 Status InferenceContext::Merge(DimensionHandle d0, DimensionHandle d1, in Merge()
[all …]
Dcommon_shape_fns.h31 DimensionHandle input_size,
34 DimensionHandle* output_size);
41 InferenceContext* c, DimensionHandle input_size,
44 DimensionHandle* output_size);
/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()
Drnn_ops.cc21 using shape_inference::DimensionHandle;
42 DimensionHandle batch_size = c->Dim(x, 0); in __anon652ace880102()
43 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anon652ace880102()
73 DimensionHandle batch_size = c->Dim(x, 0); in __anon652ace880202()
74 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anon652ace880202()
75 DimensionHandle twice_cell_size = c->Dim(w_ru, 1); in __anon652ace880202()
110 DimensionHandle batch_size = c->Dim(x, 0); in __anon652ace880302()
111 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anon652ace880302()
148 DimensionHandle batch_size = c->Dim(x, 0); in __anon652ace880402()
149 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anon652ace880402()
[all …]
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()
94 DimensionHandle channels_dim; in DecodeImageV2ShapeFn()
136 DimensionHandle last_dim; in ColorspaceShapeFn()
158 DimensionHandle unused; in NMSShapeFn()
[all …]
Dlinalg_ops.cc22 using shape_inference::DimensionHandle;
34 DimensionHandle d; in MakeBatchSquareMatrix()
59 DimensionHandle num_bands = c->Dim(lhs, -2); in BandedTriangularSolveShapeFn()
60 DimensionHandle m = c->Dim(lhs, -1); in BandedTriangularSolveShapeFn()
114 DimensionHandle m; in MatrixSolveShapeFn()
117 DimensionHandle n = c->Dim(lhs, -1); in MatrixSolveShapeFn()
146 DimensionHandle m; in MatrixTriangularSolveShapeFn()
164 DimensionHandle n; in SelfAdjointEigV2ShapeFn()
190 DimensionHandle n; in LuShapeFn()
215 DimensionHandle m = c->Dim(input, -2); in QrShapeFn()
[all …]
Dctc_ops.cc21 using shape_inference::DimensionHandle;
49 DimensionHandle unused; in __anonbca2cdf40102()
55 DimensionHandle batch_size; in __anonbca2cdf40102()
86 DimensionHandle unused; in __anonbca2cdf40202()
92 DimensionHandle batch_size; in __anonbca2cdf40202()
119 DimensionHandle batch_size; in __anonbca2cdf40302()
123 DimensionHandle total_decoded_outputs = c->UnknownDim(); in __anonbca2cdf40302()
150 DimensionHandle batch_size; in __anonbca2cdf40402()
Dset_ops.cc22 using shape_inference::DimensionHandle;
53 DimensionHandle output_rank; in __anonda6063e60102()
113 DimensionHandle input1_rank_dim = c->Dim(input1_shape_shape, 0); in __anonda6063e60202()
115 DimensionHandle output_rank_dim; in __anonda6063e60202()
162 DimensionHandle input0_rank_dim = c->Dim(input0_shape_shape, 0); in __anonda6063e60302()
163 DimensionHandle input1_rank_dim = c->Dim(input1_shape_shape, 0); in __anonda6063e60302()
164 DimensionHandle output_rank_dim; in __anonda6063e60302()
Darray_ops.cc34 using shape_inference::DimensionHandle;
67 std::vector<DimensionHandle> dims(num_dims); in PadKnown()
85 DimensionHandle unused; in PadShapeFn()
90 DimensionHandle n_dim = c->Dim(paddings, 0); in PadShapeFn()
125 DimensionHandle perm_elems = c->NumElements(perm_shape); in TransposeShapeFn()
151 std::vector<DimensionHandle> dims; in TransposeShapeFn()
203 DimensionHandle known_out_elems = c->NumElements(out); in SetOutputShapeForReshape()
207 DimensionHandle dim = c->Dim(out, i); in SetOutputShapeForReshape()
221 DimensionHandle known_in_elems = c->NumElements(in); in SetOutputShapeForReshape()
225 DimensionHandle dim = c->Dim(in, i); in SetOutputShapeForReshape()
[all …]
Dstring_ops.cc34 using shape_inference::DimensionHandle;
270 DimensionHandle pos_dim = c->Dim(pos_shape, i); in __anon4354d5f10702()
271 DimensionHandle len_dim = c->Dim(len_shape, i); in __anon4354d5f10702()
307 std::vector<DimensionHandle> dims(1); in __anon4354d5f10802()
335 DimensionHandle num_row_splits; in __anon4354d5f10902()
336 DimensionHandle input_size = c->NumElements(c->input(0)); in __anon4354d5f10902()
341 DimensionHandle num_chars = c->UnknownDim(); in __anon4354d5f10902()
358 DimensionHandle num_row_splits; in __anon4354d5f10a02()
359 DimensionHandle input_size = c->NumElements(c->input(0)); in __anon4354d5f10a02()
364 DimensionHandle num_chars = c->UnknownDim(); in __anon4354d5f10a02()
Dstateless_random_ops.cc21 using shape_inference::DimensionHandle;
29 DimensionHandle unused; in StatelessShape()
102 DimensionHandle unused_dim; in __anona1237d120202()
107 DimensionHandle num_samples; in __anona1237d120202()
142 DimensionHandle unused_dim; in __anona1237d120302()
Dlookup_ops.cc24 using shape_inference::DimensionHandle;
34 DimensionHandle unused_handle; in TwoElementVectorInputsAndScalarOutputs()
47 DimensionHandle unused_handle; in ScalarAndTwoElementVectorInputsAndScalarOutputs()
80 DimensionHandle unused_dim; in __anon18b155f40202()
142 DimensionHandle dim = c->Dim(key_shape_and_type.shape, d); in ValidateTableResourceHandle()
146 std::vector<DimensionHandle> keys_prefix_vec; in ValidateTableResourceHandle()
194 DimensionHandle unused_dim; in __anon18b155f40402()
249 DimensionHandle unused_dim; in __anon18b155f40702()
293 DimensionHandle unused_dim; in __anon18b155f40902()
451 DimensionHandle unused_dim; in __anon18b155f40e02()
[all …]
Ddata_flow_ops.cc23 using shape_inference::DimensionHandle;
163 DimensionHandle unused_handle; in TwoElementVectorInputsAndScalarOutputs()
656 DimensionHandle unused_dim; in __anonea01102b1602()
678 DimensionHandle unused_dim; in __anonea01102b1702()
694 std::vector<DimensionHandle> dims; in __anonea01102b1702()
721 DimensionHandle unused_dim; in __anonea01102b1802()
748 DimensionHandle unused_dim; in __anonea01102b1902()
774 DimensionHandle unused_dim; in __anonea01102b1a02()
809 DimensionHandle unused_dim; in __anonea01102b1b02()
838 DimensionHandle unused_dim; in __anonea01102b1c02()
[all …]
Dstateless_random_ops_v2.cc22 using shape_inference::DimensionHandle;
34 DimensionHandle unused; in StatelessShapeV2()
108 DimensionHandle unused; in __anonea62c6790202()
128 DimensionHandle unused; in __anonea62c6790302()
Dsparse_ops.cc22 using shape_inference::DimensionHandle;
96 DimensionHandle unused_dim; in __anon36a8fc880402()
111 DimensionHandle output_right = c->Dim(b, adjoint_b ? 0 : 1); in __anon36a8fc880402()
112 DimensionHandle output_left = c->Dim(a_shape, adjoint_a ? 1 : 0); in __anon36a8fc880402()
113 DimensionHandle inner_left = c->Dim(a_shape, adjoint_a ? 0 : 1); in __anon36a8fc880402()
114 DimensionHandle inner_right = c->Dim(b, adjoint_b ? 1 : 0); in __anon36a8fc880402()
161 DimensionHandle unused; in __anon36a8fc880702()
180 DimensionHandle unused; in __anon36a8fc880802()
219 DimensionHandle output_row_count = c->MakeDim(0ll); in __anon36a8fc880a02()
222 DimensionHandle output_ind_cols = c->UnknownDim(); in __anon36a8fc880a02()
[all …]
Dparsing_ops.cc24 using shape_inference::DimensionHandle;
66 const DimensionHandle& num_examples, in AddRaggedOutputShapes()
68 DimensionHandle num_splits; in AddRaggedOutputShapes()
121 DimensionHandle fixed_length; in __anonf8f6e66e0302()
129 DimensionHandle width; in __anonf8f6e66e0302()
212 DimensionHandle num_examples = c->UnknownDim(); in __anonf8f6e66e0502()
291 DimensionHandle num_examples = c->Dim(input, 0); in __anonf8f6e66e0702()
374 DimensionHandle num_examples = c->UnknownDim(); in __anonf8f6e66e0802()
Dboosted_trees_ops.cc27 using shape_inference::DimensionHandle;
326 shape_inference::DimensionHandle unused_dim; in __anon770f674c0902()
368 shape_inference::DimensionHandle batch_size = c->Dim(c->input(0), 0); in __anon770f674c0a02()
383 DimensionHandle logits_dim = c->Dim(c->input(1), 1); in __anon770f674c0a02()
384 DimensionHandle hessian_dim = c->Dim(c->input(2), 1); in __anon770f674c0a02()
385 DimensionHandle feature_dim = c->Dim(c->input(3), 1); in __anon770f674c0a02()
386 DimensionHandle stats_dim; in __anon770f674c0a02()
420 shape_inference::DimensionHandle batch_size = c->Dim(c->input(0), 0); in __anon770f674c0b02()
421 shape_inference::DimensionHandle num_entries; in __anon770f674c0b02()
439 DimensionHandle unused; in __anon770f674c0b02()
[all …]
Dnn_ops.cc29 using shape_inference::DimensionHandle;
45 std::vector<DimensionHandle> output_dims; in FractionalPoolShapeFn()
47 DimensionHandle d = c->Dim(input, i); in FractionalPoolShapeFn()
110 DimensionHandle last_dim = c->Dim(input, 3); in __anonad10f3360202()
144 DimensionHandle last_dim = c->Dim(input, 3); in __anonad10f3360302()
440 DimensionHandle new_height = c->UnknownDim(); in CommonFusedConvCalculations()
441 DimensionHandle new_width = c->UnknownDim(); in CommonFusedConvCalculations()
460 std::vector<DimensionHandle> output_dims; in CommonFusedConvCalculations()
462 DimensionHandle dim = c->Dim(resized, i); in CommonFusedConvCalculations()
491 DimensionHandle batch_size_dim = c->Dim(padded, 0); in CommonFusedConvCalculations()
[all …]
Dcandidate_sampling_ops.cc21 using shape_inference::DimensionHandle;
35 DimensionHandle batch_size = c->Dim(true_classes_shape, 0); in CandidateSamplerShapeFn()
151 DimensionHandle unused; in __anonafdb00d10202()
Dio_ops.cc23 using shape_inference::DimensionHandle;
42 DimensionHandle unused_handle; in TwoElementVectorAndScalarOutputs()
70 DimensionHandle unused_dim; in __anon52e493120202()
153 DimensionHandle unused_dim; in __anon52e493120502()
176 DimensionHandle unused_dim; in __anon52e493120602()
441 DimensionHandle unused_handle; in __anon52e493120b02()
Dragged_conversion_ops.cc23 using shape_inference::DimensionHandle;
175 DimensionHandle dense_dims = in RaggedTensorToSparseShapeFn()
179 DimensionHandle num_values = c->NumElements(rt_dense_values); in RaggedTensorToSparseShapeFn()
201 shape_inference::DimensionHandle num_rows; in RaggedTensorToVariantShapeFn()
/external/tensorflow/tensorflow/c/
Dops.cc31 using ::tensorflow::shape_inference::DimensionHandle;
133 return reinterpret_cast<TF_DimensionHandle*>(new DimensionHandle); in TF_NewDimensionHandle()
187 delete reinterpret_cast<DimensionHandle*>(handle); in TF_DeleteDimensionHandle()
231 auto* cc_result = reinterpret_cast<DimensionHandle*>(result); in TF_ShapeInferenceContextDim()
234 *cc_result = DimensionHandle(); in TF_ShapeInferenceContextDim()
245 *reinterpret_cast<DimensionHandle*>(dim_handle)); in TF_DimensionHandleValueKnown()
270 *reinterpret_cast<DimensionHandle*>(dim_handle)); in TF_DimensionHandleValue()
/external/tensorflow/tensorflow/core/tpu/ops/
Dtpu_execute_op.cc32 shape_inference::DimensionHandle unused; in __anonab1a3a9a0102()
52 shape_inference::DimensionHandle unused; in __anonab1a3a9a0202()

123