| /external/tensorflow/tensorflow/contrib/image/ops/ |
| single_image_random_dot_stereograms_ops.cc | 22 using shape_inference::DimensionHandle;
55 DimensionHandle x_dim = c->Dim(output_image_shape, 0);
56 DimensionHandle y_dim = c->Dim(output_image_shape, 1);
|
| image_ops.cc | 22 using shape_inference::DimensionHandle;
30 Status SetOutputToSizedImage(InferenceContext* c, DimensionHandle batch_dim,
31 int size_input_idx, DimensionHandle channel_dim) {
35 DimensionHandle unused;
40 DimensionHandle width;
41 DimensionHandle height;
|
| /external/tensorflow/tensorflow/contrib/rnn/ops/ |
| gru_ops.cc | 19 using tensorflow::shape_inference::DimensionHandle;
40 DimensionHandle batch_size = c->Dim(x, 0);
41 DimensionHandle cell_size = c->Dim(h_prev, 1);
119 DimensionHandle batch_size = c->Dim(x, 0);
120 DimensionHandle cell_size = c->Dim(h_prev, 1);
121 DimensionHandle twice_cell_size = c->Dim(w_ru, 1);
|
| lstm_ops.cc | 21 using shape_inference::DimensionHandle;
50 DimensionHandle batch_size = c->Dim(x, 0);
51 DimensionHandle cell_size = c->Dim(cs_prev, 1);
137 DimensionHandle batch_size = c->Dim(x, 0);
138 DimensionHandle cell_size = c->Dim(cs_prev, 1);
139 DimensionHandle cell_size_times_4;
205 DimensionHandle timelen = c->Dim(x, 0);
206 DimensionHandle batch_size = c->Dim(x, 1);
207 DimensionHandle cell_size;
|
| /external/tensorflow/tensorflow/core/framework/ |
| shape_inference.h | 56 class DimensionHandle {
58 DimensionHandle() {}
59 bool SameHandle(DimensionHandle d) const { return ptr_ == d.ptr_; }
63 DimensionHandle(const Dimension* dim) { ptr_ = dim; }
86 Shape(const std::vector<DimensionHandle>& dims);
90 const std::vector<DimensionHandle> dims_;
122 // Struct used to allow functions to take DimensionHandle or a dimension value.
127 DimensionOrConstant(DimensionHandle dim);
133 DimensionHandle dim;
342 DimensionHandle Dim(ShapeHandle s, int64 idx)
[all...] |
| common_shape_fns.cc | 140 shape_inference::DimensionHandle input_size,
143 int64 padding_after, shape_inference::DimensionHandle* output_size) {
162 DimensionHandle window_size;
187 shape_inference::DimensionHandle input_size,
189 Padding padding_type, shape_inference::DimensionHandle* output_size) {
223 DimensionHandle output_rows = transpose_a ? c->Dim(a, 1) : c->Dim(a, 0);
224 DimensionHandle output_cols = transpose_b ? c->Dim(b, 0) : c->Dim(b, 1);
227 DimensionHandle inner_a = transpose_a ? c->Dim(a, 0) : c->Dim(a, 1);
228 DimensionHandle inner_b = transpose_b ? c->Dim(b, 1) : c->Dim(b, 0);
229 DimensionHandle merged
[all...] |
| shape_inference.cc | 293 DimensionHandle dim = Dim(handle, i);
311 DimensionHandle InferenceContext::NumElements(ShapeHandle s) {
343 string InferenceContext::DebugString(DimensionHandle d) {
377 std::vector<DimensionHandle> dims;
421 Status InferenceContext::WithValue(DimensionHandle dim, int64 value,
422 DimensionHandle* out) {
429 DimensionHandle d = MakeDim(value);
437 void InferenceContext::Relax(DimensionHandle d_old, DimensionHandle d_new,
438 DimensionHandle* out)
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| audio_ops.cc | 25 using shape_inference::DimensionHandle;
33 DimensionHandle channels_dim;
45 DimensionHandle samples_dim;
78 DimensionHandle input_length = c->Dim(input, 0);
79 DimensionHandle input_channels = c->Dim(input, 1);
81 DimensionHandle output_length;
96 DimensionHandle output_channels =
113 DimensionHandle spectrogram_channels = c->Dim(spectrogram, 0);
114 DimensionHandle spectrogram_length = c->Dim(spectrogram, 1);
116 DimensionHandle output_channels = c->MakeDim(dct_coefficient_count)
[all...] |
| stateless_random_ops.cc | 21 using shape_inference::DimensionHandle;
29 DimensionHandle unused;
83 DimensionHandle unused_dim;
88 DimensionHandle num_samples;
|
| ragged_conversion_ops.cc | 21 using shape_inference::DimensionHandle;
61 DimensionHandle dense_dims =
65 DimensionHandle num_values = c->NumElements(rt_dense_values);
|
| ctc_ops.cc | 21 using shape_inference::DimensionHandle;
48 DimensionHandle unused;
54 DimensionHandle batch_size;
80 DimensionHandle batch_size;
84 DimensionHandle total_decoded_outputs = c->UnknownDim();
110 DimensionHandle batch_size;
|
| image_ops.cc | 22 using shape_inference::DimensionHandle;
30 Status SetOutputToSizedImage(InferenceContext* c, DimensionHandle batch_dim,
31 int size_input_idx, DimensionHandle channel_dim) {
35 DimensionHandle unused;
40 DimensionHandle width;
41 DimensionHandle height;
72 DimensionHandle channels_dim;
102 DimensionHandle last_dim;
124 DimensionHandle unused;
150 DimensionHandle unused
[all...] |
| ragged_math_ops.cc | 21 using shape_inference::DimensionHandle;
54 DimensionHandle dim = c->UnknownDim();
|
| tpu_cross_replica_ops.cc | 21 using shape_inference::DimensionHandle;
57 std::vector<DimensionHandle> dims;
|
| string_ops.cc | 34 using shape_inference::DimensionHandle;
246 DimensionHandle pos_dim = c->Dim(pos_shape, i);
247 DimensionHandle len_dim = c->Dim(len_shape, i);
282 std::vector<DimensionHandle> dims(1);
309 DimensionHandle num_row_splits;
310 DimensionHandle input_size = c->NumElements(c->input(0));
315 DimensionHandle num_chars = c->UnknownDim();
331 DimensionHandle num_row_splits;
332 DimensionHandle input_size = c->NumElements(c->input(0));
337 DimensionHandle num_chars = c->UnknownDim()
[all...] |
| linalg_ops.cc | 22 using shape_inference::DimensionHandle;
34 DimensionHandle d;
71 DimensionHandle m;
74 DimensionHandle n = c->Dim(lhs, -1);
93 DimensionHandle n;
119 DimensionHandle n;
144 DimensionHandle m = c->Dim(input, -2);
145 DimensionHandle n = c->Dim(input, -1);
146 DimensionHandle p;
175 DimensionHandle m = c->Dim(input, -2)
[all...] |
| set_ops.cc | 22 using shape_inference::DimensionHandle;
53 DimensionHandle output_rank;
113 DimensionHandle input1_rank_dim = c->Dim(input1_shape_shape, 0);
115 DimensionHandle output_rank_dim;
162 DimensionHandle input0_rank_dim = c->Dim(input0_shape_shape, 0);
163 DimensionHandle input1_rank_dim = c->Dim(input1_shape_shape, 0);
164 DimensionHandle output_rank_dim;
|
| candidate_sampling_ops.cc | 21 using shape_inference::DimensionHandle;
35 DimensionHandle batch_size = c->Dim(true_classes_shape, 0);
151 DimensionHandle unused;
|
| lookup_ops.cc | 24 using shape_inference::DimensionHandle;
34 DimensionHandle unused_handle;
47 DimensionHandle unused_handle;
80 DimensionHandle unused_dim;
134 DimensionHandle dim = c->Dim(key_shape_and_type.shape, d);
138 std::vector<DimensionHandle> keys_prefix_vec;
196 DimensionHandle unused_dim;
251 DimensionHandle unused_dim;
293 DimensionHandle unused_dim;
451 DimensionHandle unused_dim
[all...] |
| sparse_ops.cc | 22 using shape_inference::DimensionHandle;
96 DimensionHandle unused_dim;
111 DimensionHandle output_right = c->Dim(b, adjoint_b ? 0 : 1);
112 DimensionHandle output_left = c->Dim(a_shape, adjoint_a ? 1 : 0);
113 DimensionHandle inner_left = c->Dim(a_shape, adjoint_a ? 0 : 1);
114 DimensionHandle inner_right = c->Dim(b, adjoint_b ? 1 : 0);
161 DimensionHandle unused;
180 DimensionHandle unused;
219 DimensionHandle output_row_count = c->MakeDim(0ll);
222 DimensionHandle output_ind_cols = c->UnknownDim()
[all...] |
| /external/tensorflow/tensorflow/contrib/coder/ops/ |
| coder_ops.cc | 22 using shape_inference::DimensionHandle;
127 DimensionHandle last;
|
| /external/tensorflow/tensorflow/contrib/reduce_slice_ops/ops/ |
| reduce_slice_ops.cc | 21 using shape_inference::DimensionHandle;
29 DimensionHandle dimhandle;
30 DimensionHandle dim_axis = c->UnknownDim();
43 DimensionHandle raw_dim_axis;
|
| /external/tensorflow/tensorflow/contrib/fused_conv/ops/ |
| fused_conv2d_bias_activation_op.cc | 51 using shape_inference::DimensionHandle;
69 DimensionHandle output_depth_dim =
77 DimensionHandle bias_dim = c->Dim(bias_shape, 0);
|
| /external/tensorflow/tensorflow/contrib/seq2seq/ops/ |
| beam_search_ops.cc | 21 using shape_inference::DimensionHandle;
43 DimensionHandle batch_size = c->Dim(step_ids, 1);
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/ops/ |
| model_ops.cc | 21 using shape_inference::DimensionHandle;
95 DimensionHandle num_points = c->UnknownDim();
130 DimensionHandle num_points = c->UnknownDim();
|
