/external/tensorflow/tensorflow/core/framework/ |
D | shape_inference.h | 56 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 …]
|
D | common_shape_fns.cc | 140 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 …]
|
D | shape_inference.cc | 293 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/ |
D | audio_ops.cc | 25 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()
|
D | image_ops.cc | 22 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 …]
|
D | linalg_ops.cc | 22 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 …]
|
D | string_ops.cc | 34 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()
|
D | array_ops.cc | 27 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 …]
|
D | ctc_ops.cc | 21 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()
|
D | set_ops.cc | 22 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()
|
D | lookup_ops.cc | 24 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 …]
|
D | stateless_random_ops.cc | 21 using shape_inference::DimensionHandle; 29 DimensionHandle unused; in StatelessShape() 83 DimensionHandle unused_dim; in __anon530aab340202() 88 DimensionHandle num_samples; in __anon530aab340202()
|
D | sparse_ops.cc | 22 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 …]
|
D | data_flow_ops.cc | 23 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 …]
|
D | nn_ops.cc | 26 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 …]
|
D | ragged_conversion_ops.cc | 21 using shape_inference::DimensionHandle; 61 DimensionHandle dense_dims = in RaggedTensorToSparseShapeFn() 65 DimensionHandle num_values = c->NumElements(rt_dense_values); in RaggedTensorToSparseShapeFn()
|
D | candidate_sampling_ops.cc | 21 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/ |
D | lstm_ops.cc | 21 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()
|
D | gru_ops.cc | 19 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/ |
D | image_ops.cc | 22 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()
|
D | single_image_random_dot_stereograms_ops.cc | 22 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/ |
D | reduce_slice_ops.cc | 21 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/ |
D | fused_conv2d_bias_activation_op.cc | 51 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/ |
D | split_handler_ops.cc | 21 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/ |
D | model_ops.cc | 21 using shape_inference::DimensionHandle; 95 DimensionHandle num_points = c->UnknownDim(); in __anon5b1697710102() 130 DimensionHandle num_points = c->UnknownDim(); in __anon5b1697710202()
|