Home
last modified time | relevance | path

Searched refs:values_shape (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dweights_broadcast_ops.py33 def _has_valid_dims(weights_shape, values_shape): argument
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): argument
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 …]
Dwhile_v2_indexed_slices_rewriter.py176 values_shape = tensor_shape.TensorShape([0] +
178 values = array_ops.zeros(values_shape, dtype=values_out.dtype,
185 values_shape = array_ops.concat([[0], forward_shape[1:]], 0)
186 values = array_ops.zeros(values_shape, dtype=values_out.dtype,
Darray_grad.py408 values_shape = array_ops.concat([size, params_shape[1:]], 0)
409 values = array_ops.reshape(grad, values_shape)
439 values_shape = array_ops.concat([indices_size, params_tail_shape], 0)
440 values = array_ops.reshape(grad, values_shape)
453 values_shape = array_ops.concat([outer_shape, indices_size, inner_shape], 0)
454 values = array_ops.reshape(grad, values_shape)
Dcontrol_flow_ops.py2759 values_shape = tensor_shape.TensorShape([tensor_shape.Dimension(1)] +
2764 0, values.dtype, shape=values_shape, name="b_acc")
2768 values_shape = _resource_safe_shape(op.inputs[0])[1:]
2769 values_shape = array_ops.concat([[1], values_shape], 0)
2770 values_acc = array_ops.zeros(values_shape, dtype=values.dtype)
Dresource_variable_ops.py1516 values_shape = array_ops.concat([size, params_shape[1:]], 0)
1517 values = array_ops.reshape(grad, values_shape)
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor.py651 values_shape = self._values.shape
652 value_shape = values_shape[1:]
1218 values_shape = array_ops.concat([[nvals], input_shape[2:]], axis=0)
1219 values = array_ops.reshape(tensor, values_shape)
1272 values_shape = array_ops.shape(values, out_type=dtypes.int64)
1273 value_shape = values_shape[1:]
1274 nvals = values_shape[0]
1485 values_shape = tensor_shape.TensorShape([None]).concatenate(value_shape)
1486 return ((values_shape, self._row_splits.shape) +
1501 values_shape = tensor_shape.TensorShape([None]).concatenate(value_shape)
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_sparse_kernel.cc189 const TensorShape& values_shape) { in MakeIndexSuffixes() argument
191 for (int dim = 1; dim < values_shape.dims(); ++dim) { in MakeIndexSuffixes()
194 for (int i = 0; i < values_shape.dim_size(dim); ++i) { in MakeIndexSuffixes()
Dragged_gather_op.cc235 TensorShape values_shape = params_dense_values_in.shape(); in WriteValues() local
236 values_shape.set_dim(0, num_values); in WriteValues()
238 context->allocate_output(values_index, values_shape, &values_out)); in WriteValues()
Dexample_parsing_ops.cc696 TensorShape values_shape({0}); in Compute() local
703 ctx, context_sparse_values.allocate(d, values_shape, &sp_values_d)); in Compute()
806 TensorShape values_shape({total_num_features}); in Compute() local
812 OP_REQUIRES_OK(ctx, feature_list_sparse_values.allocate(d, values_shape, in Compute()
/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.cc1101 TensorShape values_shape; in FastParseExample() local
1102 values_shape.AddDim(total_num_features); in FastParseExample()
1103 result->sparse_values.emplace_back(config.sparse[d].dtype, values_shape); in FastParseExample()
1176 TensorShape values_shape; in FastParseExample() local
1179 values_shape.AddDim(batch_size); in FastParseExample()
1180 values_shape.AddDim(max_num_elements); in FastParseExample()
1182 values_shape.AddDim(config.dense[d].shape.dim_size(i)); in FastParseExample()
1184 Tensor values(config.dense[d].dtype, values_shape); in FastParseExample()
1271 TensorShape values_shape; in FastParseSingleExample() local
1272 if (!config.dense[d].shape.AsTensorShape(&values_shape)) { in FastParseSingleExample()
[all …]
Dexample_proto_helper.cc312 output_shapes->values_shape.AddDim(total_num_features); in GetSparseTensorShapes()
379 const TensorShape& values_shape = sparse_tensor_batch_shapes.values_shape; in BatchExampleProtoToTensors() local
384 (*output_sparse_values_tensor)[d] = Tensor(allocator, dtype, values_shape); in BatchExampleProtoToTensors()
Dexample_proto_helper.h87 TensorShape values_shape; member
/external/tensorflow/tensorflow/lite/kernels/
Dsparse_to_dense_test.cc33 std::initializer_list<int> values_shape, T default_value, in SparseToDenseOpModel() argument
45 BuildInterpreter({indices_shape, output_shape_shape, values_shape, {1}}); in SparseToDenseOpModel()
/external/tensorflow/tensorflow/python/framework/
Dsparse_tensor.py131 values_shape = values.get_shape().with_rank(1)
135 indices_shape.dims[0].merge_with(values_shape.dims[0])
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dembedding_ops.py321 values_shape = array_ops.shape(values)
337 expected_shape = array_ops.concat([values_shape, dimension_tensor], 0)
342 " does not match the shape of values: ", values_shape])]):
377 array_ops.concat([values_shape, [dimension]], 0))
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.h318 ShapeHandle values_shape, ShapeHandle shape_shape);
Dcommon_shape_fns.cc1495 ShapeHandle values_shape, ShapeHandle shape_shape) { in ValidateSparseTensor() argument
1499 TF_RETURN_IF_ERROR(c->WithRank(values_shape, 1, &unused_shape)); in ValidateSparseTensor()
1505 DimensionHandle num_values_elements_dim = c->Dim(values_shape, 0); in ValidateSparseTensor()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc672 Shape values_shape = ShapeUtil::MakeShape(F32, {2, 3}); in TEST_F() local
676 auto values = Parameter(&builder, 2, values_shape, "values"); in TEST_F()
Dtuple_points_to_analysis_test.cc1090 Shape values_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
1094 HloInstruction::CreateParameter(1, values_shape, "values")); in TEST_F()
1097 MakeSortHlo(ShapeUtil::MakeTupleShape({keys_shape, values_shape}), in TEST_F()
Dhlo_dataflow_analysis_test.cc2469 Shape values_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2473 HloInstruction::CreateParameter(1, values_shape, "values")); in TEST_F()
2476 MakeSortHlo(ShapeUtil::MakeTupleShape({keys_shape, values_shape}), in TEST_F()
Dalgebraic_simplifier_test.cc2821 Shape values_shape = ShapeUtil::MakeShape(S32, {5, 0}); in TEST_F() local
2825 HloInstruction::CreateParameter(1, values_shape, "values0")); in TEST_F()
2827 HloInstruction::CreateParameter(2, values_shape, "values1")); in TEST_F()
2829 {keys_shape, values_shape, values_shape}), in TEST_F()
/external/tensorflow/tensorflow/contrib/metrics/python/ops/
Dmetric_ops.py3605 values_shape = values.get_shape()
3606 if values_shape.dims is None:
3609 ndim = len(values_shape)
3615 fixed_shape = [dim.value for n, dim in enumerate(values_shape) if n != axis]