| /external/tensorflow/tensorflow/python/ops/ |
| weights_broadcast_ops.py | 33 def _has_valid_dims(weights_shape, values_shape):
35 None, "has_invalid_dims", (weights_shape, values_shape)) as scope:
36 values_shape_2d = array_ops.expand_dims(values_shape, -1)
47 weights_rank, weights_shape, values_rank, values_shape):
50 (weights_rank, weights_shape, values_rank, values_shape)) as scope:
55 lambda: _has_valid_dims(weights_shape, values_shape),
90 values_shape = array_ops.shape(values, name="shape")
105 values_shape_static = tensor_util.constant_value(values_shape)
124 "values.shape=", values.name, values_shape,
131 weights_rank, weights_shape, values_rank, values_shape),
[all...] |
| array_grad.py | 406 values_shape = array_ops.concat([size, params_shape[1:]], 0)
407 values = array_ops.reshape(grad, values_shape)
437 values_shape = array_ops.concat([indices_size, params_tail_shape], 0)
438 values = array_ops.reshape(grad, values_shape)
451 values_shape = array_ops.concat([outer_shape, indices_size, inner_shape], 0)
452 values = array_ops.reshape(grad, values_shape)
|
| resource_variable_ops.py | [all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| sparse_tensor.py | 134 values_shape = values.get_shape().with_rank(1)
138 indices_shape[0].merge_with(values_shape[0])
|
| /external/tensorflow/tensorflow/core/util/ |
| example_proto_fast_parsing.cc | 1170 TensorShape values_shape; local
1367 TensorShape values_shape; local
[all...] |
| example_proto_helper.cc | 312 output_shapes->values_shape.AddDim(total_num_features);
379 const TensorShape& values_shape = sparse_tensor_batch_shapes.values_shape; local
384 (*output_sparse_values_tensor)[d] = Tensor(allocator, dtype, values_shape);
|
| example_proto_helper.h | 86 TensorShape values_shape; member in struct:tensorflow::VarLenFeatureBatchShapes
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| embedding_ops.py | 307 values_shape = array_ops.shape(values)
323 expected_shape = array_ops.concat([values_shape, dimension_tensor], 0)
328 " does not match the shape of values: ", values_shape])]):
363 array_ops.concat([values_shape, [dimension]], 0))
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| common_shape_fns.h | 278 ShapeHandle values_shape, ShapeHandle shape_shape);
|
| common_shape_fns.cc | [all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| example_parsing_ops.cc | 522 TensorShape values_shape({0});
529 ctx, context_sparse_values.allocate(d, values_shape, &sp_values_d));
632 TensorShape values_shape({total_num_features});
638 OP_REQUIRES_OK(ctx, feature_list_sparse_values.allocate(d, values_shape,
|
| /external/tensorflow/tensorflow/contrib/metrics/python/ops/ |
| metric_ops.py | [all...] |
