| /external/tensorflow/tensorflow/contrib/specs/python/ |
| summaries_test.py | 38 inputs_shape = (1, 18, 19, 5)
39 inputs = constant_op.constant(_rand(*inputs_shape))
47 spec, input_shape=inputs_shape),
|
| /external/tensorflow/tensorflow/core/kernels/ |
| ctc_decoder_ops.cc | 66 const TensorShape& inputs_shape = (*inputs)->shape(); local
68 if (inputs_shape.dims() != 3) {
72 const int64 max_time = inputs_shape.dim_size(0);
73 const int64 batch_size = inputs_shape.dim_size(1);
190 const TensorShape& inputs_shape = inputs->shape(); variable
193 const int64 max_time = inputs_shape.dim_size(0);
194 const int64 batch_size = inputs_shape.dim_size(1);
195 const int64 num_classes_raw = inputs_shape.dim_size(2);
282 const TensorShape& inputs_shape = inputs->shape(); variable
284 const int64 max_time = inputs_shape.dim_size(0)
[all...] |
| ctc_loss_op.cc | 68 const TensorShape& inputs_shape = inputs->shape(); variable
69 const int64 max_time = inputs_shape.dim_size(0);
70 const int64 batch_size = inputs_shape.dim_size(1);
71 const int64 num_classes_raw = inputs_shape.dim_size(2);
144 ctx->allocate_output("gradient", inputs_shape, &gradient));
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| rnn_cells.py | 91 def build(self, inputs_shape):
93 super(MaskedBasicLSTMCell, self).build(inputs_shape)
97 input_depth = inputs_shape.dims[1].value
240 def build(self, inputs_shape):
242 super(MaskedLSTMCell, self).build(inputs_shape)
246 input_depth = inputs_shape.dims[1].value
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| lstm_ops.py | 399 def build(self, inputs_shape):
400 if not inputs_shape.dims[1].value:
402 "Expecting inputs_shape[1] to be set: %s" % str(inputs_shape))
403 input_size = inputs_shape.dims[1].value
516 inputs_shape = inputs.get_shape().with_rank(3)
517 if not inputs_shape[2]:
518 raise ValueError("Expecting inputs_shape[2] to be set: %s" % inputs_shape)
519 batch_size = inputs_shape.dims[1].valu
[all...] |
| rnn_cell.py | [all...] |
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| rnn_cell.py | 94 def build(self, inputs_shape):
98 inputs_shape: Rnn input tensor shape.
103 if inputs_shape[-1] is None:
105 (inputs_shape,))
107 input_depth = inputs_shape[-1]
284 def build(self, inputs_shape):
288 inputs_shape: The inputs_shape must be known, and is [batch_size,
292 ValueError: if the inputs_shape is invalid.
294 if len(inputs_shape) != 2
[all...] |
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| normalization.py | 94 inputs_shape = inputs.shape
110 [1, inputs_shape[1].value] + [1 for _ in range(2, inputs_rank)])
117 params_shape = inputs_shape[reduction_axis:reduction_axis + 1]
309 inputs_shape = (axes_before_channels + [groups, channels // groups] +
311 inputs = array_ops.reshape(inputs, inputs_shape)
|
| layers.py | 285 inputs_shape = inputs.get_shape()
287 params_shape = inputs_shape[-1:]
289 params_shape = inputs_shape[1:2]
421 outputs.set_shape(inputs_shape)
677 inputs_shape = inputs.get_shape()
678 inputs_rank = inputs_shape.ndims
684 inputs_shape[0:1].assert_is_compatible_with(batch_weights.get_shape())
691 params_shape = inputs_shape[1:2]
696 [1, inputs_shape.dims[1].value] + [1 for _ in range(2, inputs_rank)])
699 params_shape = inputs_shape[-1:
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| rnn_cell_impl.py | 444 def build(self, inputs_shape):
445 if inputs_shape[-1] is None:
447 % str(inputs_shape))
450 input_depth = inputs_shape[-1]
545 def build(self, inputs_shape):
546 if inputs_shape[-1] is None:
548 % str(inputs_shape))
550 input_depth = inputs_shape[-1]
724 def build(self, inputs_shape):
725 if inputs_shape[-1] is None
[all...] |
| /external/tensorflow/tensorflow/python/keras/layers/ |
| convolutional.py | [all...] |
| recurrent_test.py | [all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| cudnn_rnn_ops.py | 118 def build(self, inputs_shape):
119 if inputs_shape[1].value is None:
121 % inputs_shape)
123 input_depth = inputs_shape[1].value
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| arithmetic_optimizer_test.cc | 897 Output inputs_shape = ops::Shape(s, inputs); local
932 Output inputs_shape = ops::Shape(s, inputs); local
1359 Output inputs_shape = local
1392 Output inputs_shape = local
1469 Output inputs_shape = local
1493 Output inputs_shape = local
[all...] |
