/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();
|