| /external/tensorflow/tensorflow/core/kernels/ |
| D | nth_element_op.cc | 95 const int last_dim = input_tensor.dim_size(input_tensor.dims() - 1); in operator ()() local
98 auto SubNthElement = [&, input, output, last_dim, n](int start, int limit) { in operator ()()
100 std::vector<T> buf(last_dim); in operator ()()
104 const T* input_start = input + b * last_dim; in operator ()()
105 const T* input_end = input + (b + 1) * last_dim; in operator ()()
120 20 * last_dim, SubNthElement); in operator ()()
|
| D | quantized_bias_add_op.cc | 52 const auto last_dim = input.shape().dims() - 1; in Compute() local
54 context, bias.shape().dim_size(0) == input.shape().dim_size(last_dim), in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | sorting.cc | 28 int last_dim = input_shape.dimensions_size() - 1; in TopK() local
32 XlaOp iota_s32 = Iota(builder, iota_shape, last_dim); in TopK()
40 last_dim, /*is_stable=*/true); in TopK()
43 limit_indices[last_dim] = k; in TopK()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | topk_op.cc | 42 int last_dim = input_shape.dims() - 1; in Compile() local
43 int last_dim_size = input_shape.dim_size(last_dim); in Compile()
|
| D | diag_op.cc | 178 int last_dim = dims.size() - 1; in Compile() local
179 int64 last_dim_size = input_shape.dim_size(last_dim); in Compile()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | nn_ops.cc | 113 DimensionHandle last_dim = c->Dim(input, 3); in __anon3e672dd80302() local
117 TF_RETURN_IF_ERROR(c->Merge(last_dim, c->Dim(vec, 0), &last_dim)); in __anon3e672dd80302()
121 TF_RETURN_IF_ERROR(c->ReplaceDim(input, 3, last_dim, &out)); in __anon3e672dd80302()
147 DimensionHandle last_dim = c->Dim(input, 3); in __anon3e672dd80402() local
151 TF_RETURN_IF_ERROR(c->Merge(last_dim, c->Dim(vec, 0), &last_dim)); in __anon3e672dd80402()
155 TF_RETURN_IF_ERROR(c->ReplaceDim(input, 3, last_dim, &dx)); in __anon3e672dd80402()
158 ShapeHandle vector_shape = c->Vector(last_dim); in __anon3e672dd80402()
1216 DimensionHandle last_dim = c->Dim(input, -1); in TopKShapeFn() local
1217 if (c->ValueKnown(last_dim) && c->ValueKnown(k_dim) && in TopKShapeFn()
1218 c->Value(last_dim) < c->Value(k_dim)) { in TopKShapeFn()
[all …]
|
| D | image_ops.cc | 102 DimensionHandle last_dim; in ColorspaceShapeFn() local
103 TF_RETURN_IF_ERROR(c->WithValue(c->Dim(input, -1), 3, &last_dim)); in ColorspaceShapeFn()
105 TF_RETURN_IF_ERROR(c->ReplaceDim(input, -1, last_dim, &out)); in ColorspaceShapeFn()
|
| D | array_ops.cc | 2658 DimensionHandle last_dim = c->Dim(new_shape, -1); in __anon7c94107b3b02() local
2659 if (!c->ValueKnown(last_dim) || c->Value(last_dim) == divisor_val) { in __anon7c94107b3b02()
2663 c->Value(last_dim), " does not match ", in __anon7c94107b3b02()
3124 ShapeHandle last_dim = c->Vector(c->Dim(inputs, -1)); in __anon7c94107b4a02() local
3128 TF_RETURN_IF_ERROR(c->Merge(min_max, last_dim, &min_max)); in __anon7c94107b4a02()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | core.py | 988 last_dim = tensor_shape.dimension_value(input_shape[-1])
990 axes={-1: last_dim})
993 shape=[last_dim, self.units],
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | layers_test.py | 3978 for last_dim in [1, 3]:
3980 self._fully_connected_basic_use(self.input_3_dim, last_dim,
3981 [2, 4, last_dim])
4172 def _unknown_dim_invalid_input(self, last_dim): argument
4173 x = array_ops.placeholder(dtypes.float32, shape=[3, last_dim])
4177 self._unknown_dim_invalid_input(last_dim=3)
4182 self._unknown_dim_invalid_input(last_dim=None)
|