Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/framework/
Dshape_inference.h129 class DimensionHandle {
131 DimensionHandle() {} in DimensionHandle() function
132 bool SameHandle(DimensionHandle d) const { return ptr_ == d.ptr_; } in SameHandle()
136 DimensionHandle(const Dimension* dim) { ptr_ = dim; } in DimensionHandle() function
157 Shape(const std::vector<DimensionHandle>& dims);
161 const std::vector<DimensionHandle> dims_;
195 DimensionOrConstant(DimensionHandle dim);
201 DimensionHandle dim;
413 DimensionHandle Dim(ShapeHandle s, int64_t idx) { in Dim()
420 static DimensionHandle DimKnownRank(ShapeHandle s, int64_t idx) { in DimKnownRank()
[all …]
Dcommon_shape_fns.cc37 shape_inference::DimensionHandle input_size, in GetWindowedOutputSizeFromDimsV2()
40 int64_t 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.cc243 DimensionHandle dim = Dim(handle, i); in ShapeHandleToProto()
261 DimensionHandle InferenceContext::NumElements(ShapeHandle s) { in NumElements()
293 string InferenceContext::DebugString(DimensionHandle d) { in DebugString()
326 std::vector<DimensionHandle> dims; in WithRank()
370 Status InferenceContext::WithValue(DimensionHandle dim, int64_t value, in WithValue()
371 DimensionHandle* out) { in WithValue()
378 DimensionHandle d = MakeDim(value); in WithValue()
386 void InferenceContext::Relax(DimensionHandle d_old, DimensionHandle d_new, in Relax()
387 DimensionHandle* out) { in Relax()
412 Status InferenceContext::Merge(DimensionHandle d0, DimensionHandle d1, in Merge()
[all …]
/external/tensorflow/tensorflow/core/ops/
Duniform_quant_ops.cc24 using shape_inference::DimensionHandle;
29 DimensionHandle match_dimension_handle, in ScalesZeroPointsShapeValid()
46 DimensionHandle scales_size = context->Dim(scales, 0); in ScalesZeroPointsShapeValid()
47 DimensionHandle zero_points_size = context->Dim(zero_points, 0); in ScalesZeroPointsShapeValid()
48 DimensionHandle merged_scales; in ScalesZeroPointsShapeValid()
51 DimensionHandle merged_zero_points; in ScalesZeroPointsShapeValid()
70 DimensionHandle inner_lhs = context->Dim(lhs, 1); in DotHybridShape()
71 DimensionHandle inner_rhs = context->Dim(rhs, 0); in DotHybridShape()
72 DimensionHandle merged; in DotHybridShape()
75 DimensionHandle output_rows = context->Dim(lhs, 0); in DotHybridShape()
[all …]
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 __anona90d3fa60102()
43 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anona90d3fa60102()
73 DimensionHandle batch_size = c->Dim(x, 0); in __anona90d3fa60202()
74 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anona90d3fa60202()
75 DimensionHandle twice_cell_size = c->Dim(w_ru, 1); in __anona90d3fa60202()
110 DimensionHandle batch_size = c->Dim(x, 0); in __anona90d3fa60302()
111 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anona90d3fa60302()
148 DimensionHandle batch_size = c->Dim(x, 0); in __anona90d3fa60402()
149 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anona90d3fa60402()
[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 __anon00853f120102()
55 DimensionHandle batch_size; in __anon00853f120102()
86 DimensionHandle unused; in __anon00853f120202()
92 DimensionHandle batch_size; in __anon00853f120202()
120 DimensionHandle batch_size; in __anon00853f120302()
124 DimensionHandle total_decoded_outputs = c->UnknownDim(); in __anon00853f120302()
151 DimensionHandle batch_size; in __anon00853f120402()
Darray_ops.cc35 using shape_inference::DimensionHandle;
68 std::vector<DimensionHandle> dims(num_dims); in PadKnown()
86 DimensionHandle unused; in PadShapeFn()
91 DimensionHandle n_dim = c->Dim(paddings, 0); in PadShapeFn()
126 DimensionHandle perm_elems = c->NumElements(perm_shape); in TransposeShapeFn()
152 std::vector<DimensionHandle> dims; in TransposeShapeFn()
204 DimensionHandle known_out_elems = c->NumElements(out); in SetOutputShapeForReshape()
208 DimensionHandle dim = c->Dim(out, i); in SetOutputShapeForReshape()
222 DimensionHandle known_in_elems = c->NumElements(in); in SetOutputShapeForReshape()
226 DimensionHandle dim = c->Dim(in, i); in SetOutputShapeForReshape()
[all …]
Dstring_ops.cc34 using shape_inference::DimensionHandle;
283 DimensionHandle pos_dim = c->Dim(pos_shape, i); in __anond1fe1a4f0702()
284 DimensionHandle len_dim = c->Dim(len_shape, i); in __anond1fe1a4f0702()
320 std::vector<DimensionHandle> dims(1); in __anond1fe1a4f0802()
348 DimensionHandle num_row_splits; in __anond1fe1a4f0902()
349 DimensionHandle input_size = c->NumElements(c->input(0)); in __anond1fe1a4f0902()
354 DimensionHandle num_chars = c->UnknownDim(); in __anond1fe1a4f0902()
371 DimensionHandle num_row_splits; in __anond1fe1a4f0a02()
372 DimensionHandle input_size = c->NumElements(c->input(0)); in __anond1fe1a4f0a02()
377 DimensionHandle num_chars = c->UnknownDim(); in __anond1fe1a4f0a02()
Dset_ops.cc22 using shape_inference::DimensionHandle;
53 DimensionHandle output_rank; in __anon1e42d5040102()
113 DimensionHandle input1_rank_dim = c->Dim(input1_shape_shape, 0); in __anon1e42d5040202()
115 DimensionHandle output_rank_dim; in __anon1e42d5040202()
162 DimensionHandle input0_rank_dim = c->Dim(input0_shape_shape, 0); in __anon1e42d5040302()
163 DimensionHandle input1_rank_dim = c->Dim(input1_shape_shape, 0); in __anon1e42d5040302()
164 DimensionHandle output_rank_dim; in __anon1e42d5040302()
Dlookup_ops.cc24 using shape_inference::DimensionHandle;
34 DimensionHandle unused_handle; in TwoElementVectorInputsAndScalarOutputs()
47 DimensionHandle unused_handle; in ScalarAndTwoElementVectorInputsAndScalarOutputs()
80 DimensionHandle unused_dim; in __anona75a9a520202()
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 __anona75a9a520402()
249 DimensionHandle unused_dim; in __anona75a9a520702()
293 DimensionHandle unused_dim; in __anona75a9a520902()
312 DimensionHandle unused; in __anona75a9a520a02()
[all …]
Dstateless_random_ops.cc21 using shape_inference::DimensionHandle;
29 DimensionHandle unused; in StatelessShape()
102 DimensionHandle unused_dim; in __anonef3c4ef00202()
107 DimensionHandle num_samples; in __anonef3c4ef00202()
142 DimensionHandle unused_dim; in __anonef3c4ef00302()
Ddata_flow_ops.cc23 using shape_inference::DimensionHandle;
163 DimensionHandle unused_handle; in TwoElementVectorInputsAndScalarOutputs()
656 DimensionHandle unused_dim; in __anon89875fc91602()
678 DimensionHandle unused_dim; in __anon89875fc91702()
694 std::vector<DimensionHandle> dims; in __anon89875fc91702()
721 DimensionHandle unused_dim; in __anon89875fc91802()
748 DimensionHandle unused_dim; in __anon89875fc91902()
774 DimensionHandle unused_dim; in __anon89875fc91a02()
809 DimensionHandle unused_dim; in __anon89875fc91b02()
838 DimensionHandle unused_dim; in __anon89875fc91c02()
[all …]
Dstateless_random_ops_v2.cc22 using shape_inference::DimensionHandle;
34 DimensionHandle unused; in StatelessShapeV2()
117 DimensionHandle unused; in __anon148fb3970202()
136 DimensionHandle unused; in __anon148fb3970302()
Dsparse_ops.cc24 using shape_inference::DimensionHandle;
98 DimensionHandle unused_dim; in __anonc55240e60402()
113 DimensionHandle output_right = c->Dim(b, adjoint_b ? 0 : 1); in __anonc55240e60402()
114 DimensionHandle output_left = c->Dim(a_shape, adjoint_a ? 1 : 0); in __anonc55240e60402()
115 DimensionHandle inner_left = c->Dim(a_shape, adjoint_a ? 0 : 1); in __anonc55240e60402()
116 DimensionHandle inner_right = c->Dim(b, adjoint_b ? 1 : 0); in __anonc55240e60402()
165 DimensionHandle unused; in __anonc55240e60702()
184 DimensionHandle unused; in __anonc55240e60802()
223 DimensionHandle output_row_count = c->MakeDim(0ll); in __anonc55240e60a02()
226 DimensionHandle output_ind_cols = c->UnknownDim(); in __anonc55240e60a02()
[all …]
Dboosted_trees_ops.cc27 using shape_inference::DimensionHandle;
326 shape_inference::DimensionHandle unused_dim; in __anon0a5055ea0902()
368 shape_inference::DimensionHandle batch_size = c->Dim(c->input(0), 0); in __anon0a5055ea0a02()
383 DimensionHandle logits_dim = c->Dim(c->input(1), 1); in __anon0a5055ea0a02()
384 DimensionHandle hessian_dim = c->Dim(c->input(2), 1); in __anon0a5055ea0a02()
385 DimensionHandle feature_dim = c->Dim(c->input(3), 1); in __anon0a5055ea0a02()
386 DimensionHandle stats_dim; in __anon0a5055ea0a02()
420 shape_inference::DimensionHandle batch_size = c->Dim(c->input(0), 0); in __anon0a5055ea0b02()
421 shape_inference::DimensionHandle num_entries; in __anon0a5055ea0b02()
439 DimensionHandle unused; in __anon0a5055ea0b02()
[all …]
Dparsing_ops.cc24 using shape_inference::DimensionHandle;
66 const DimensionHandle& num_examples, in AddRaggedOutputShapes()
68 DimensionHandle num_splits; in AddRaggedOutputShapes()
123 DimensionHandle fixed_length; in __anon5cc8b68c0302()
131 DimensionHandle width; in __anon5cc8b68c0302()
214 DimensionHandle num_examples = c->UnknownDim(); in __anon5cc8b68c0502()
293 DimensionHandle num_examples = c->Dim(input, 0); in __anon5cc8b68c0702()
376 DimensionHandle num_examples = c->UnknownDim(); in __anon5cc8b68c0802()
Dnn_ops.cc32 using shape_inference::DimensionHandle;
48 std::vector<DimensionHandle> output_dims; in FractionalPoolShapeFn()
50 DimensionHandle d = c->Dim(input, i); in FractionalPoolShapeFn()
114 DimensionHandle last_dim = c->Dim(input, 3); in __anon1bbab8940202()
148 DimensionHandle last_dim = c->Dim(input, 3); in __anon1bbab8940302()
474 DimensionHandle new_height = c->UnknownDim(); in CommonFusedConvCalculations()
475 DimensionHandle new_width = c->UnknownDim(); in CommonFusedConvCalculations()
494 std::vector<DimensionHandle> output_dims; in CommonFusedConvCalculations()
496 DimensionHandle dim = c->Dim(resized, i); in CommonFusedConvCalculations()
525 DimensionHandle batch_size_dim = c->Dim(padded, 0); in CommonFusedConvCalculations()
[all …]
Dtpu_cross_replica_ops.cc21 using shape_inference::DimensionHandle;
54 DimensionHandle num_replicas_per_group = c->Dim(group_assignment, 1); in __anond991e9790102()
82 std::vector<DimensionHandle> dims; in __anond991e9790102()
Dcandidate_sampling_ops.cc21 using shape_inference::DimensionHandle;
35 DimensionHandle batch_size = c->Dim(true_classes_shape, 0); in CandidateSamplerShapeFn()
151 DimensionHandle unused; in __anone76fc22f0202()
Dio_ops.cc23 using shape_inference::DimensionHandle;
42 DimensionHandle unused_handle; in TwoElementVectorAndScalarOutputs()
70 DimensionHandle unused_dim; in __anonc18e58700202()
159 DimensionHandle unused_dim; in __anonc18e58700502()
182 DimensionHandle unused_dim; in __anonc18e58700602()
447 DimensionHandle unused_handle; in __anonc18e58700b02()
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_meta_ops.cc77 std::vector<shape_inference::DimensionHandle> out_dims; in __anon82f11ae70102()
80 shape_inference::DimensionHandle dim = c->Dim(in, i); in __anon82f11ae70102()
86 shape_inference::DimensionHandle out_dim; in __anon82f11ae70102()
134 std::vector<shape_inference::DimensionHandle> out_dims; in __anon82f11ae70202()
137 shape_inference::DimensionHandle dim = c->Dim(in, i); in __anon82f11ae70202()
143 shape_inference::DimensionHandle out_dim; in __anon82f11ae70202()
/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()

123