Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/
Dweights_broadcast_ops.py34 def _has_valid_dims(weights_shape, values_shape): argument
36 None, "has_invalid_dims", (weights_shape, values_shape)) as scope:
37 values_shape_2d = array_ops.expand_dims(values_shape, -1)
48 weights_rank, weights_shape, values_rank, values_shape): argument
51 (weights_rank, weights_shape, values_rank, values_shape)) as scope:
56 lambda: _has_valid_dims(weights_shape, values_shape),
91 values_shape = array_ops.shape(values, name="shape")
106 values_shape_static = tensor_util.constant_value(values_shape)
125 "values.shape=", values.name, values_shape,
132 weights_rank, weights_shape, values_rank, values_shape),
[all …]
Dwhile_v2_indexed_slices_rewriter.py185 values_shape = tensor_shape.TensorShape([0] +
187 values = array_ops.zeros(values_shape, dtype=values_out.dtype,
194 values_shape = array_ops.concat([[0], forward_shape[1:]], 0)
195 values = array_ops.zeros(values_shape, dtype=values_out.dtype,
Darray_grad.py578 values_shape = array_ops.concat([size, params_shape[1:]], 0)
580 _IndexedSlicesToTensorNoWarning(grad), values_shape)
613 values_shape = array_ops.shape(values)
615 outer_shape = values_shape[:batch_dims]
616 inner_shape = values_shape[batch_dims:][1:]
666 values_shape = array_ops.concat([indices_size, params_tail_shape], 0)
668 _IndexedSlicesToTensorNoWarning(grad), values_shape)
676 values_shape = array_ops.concat([outer_shape, [-1], inner_shape], 0)
678 values_dims = array_ops.size(values_shape)
686 _IndexedSlicesToTensorNoWarning(grad), values_shape)
Dcontrol_flow_ops.py2059 values_shape = tensor_shape.TensorShape([tensor_shape.Dimension(1)] +
2064 0, values.dtype, shape=values_shape, name="b_acc")
2068 values_shape = _resource_safe_shape(op.inputs[0])[1:]
2069 values_shape = array_ops.concat([[1], values_shape], 0)
2070 values_acc = array_ops.zeros(values_shape, dtype=values.dtype)
Dresource_variable_ops.py2162 values_shape = array_ops.concat([size, params_shape[1:]], 0)
2163 values = array_ops.reshape(grad, values_shape)
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dlower_upper_bound_ops.cc36 const TensorShape values_shape = ctx->InputShape("values"); in BuildLowerUpperBoundOp() local
44 OP_REQUIRES(ctx, values_shape.dims() == 2, in BuildLowerUpperBoundOp()
49 auto new_values_shape = values_shape; in BuildLowerUpperBoundOp()
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_variant_op.cc52 TensorShape values_shape = batched_ragged.values().shape(); in UnbatchRaggedZerothDim() local
60 values_shape.set_dim(0, num_values); in UnbatchRaggedZerothDim()
62 Tensor(DataTypeToEnum<VALUE_TYPE>::value, values_shape)); in UnbatchRaggedZerothDim()
114 values_shape.set_dim(0, num_values); in UnbatchRaggedZerothDim()
116 Tensor(DataTypeToEnum<VALUE_TYPE>::value, values_shape)); in UnbatchRaggedZerothDim()
Dragged_tensor_to_sparse_kernel.cc191 const TensorShape& values_shape) { in MakeIndexSuffixes() argument
193 for (int dim = 1; dim < values_shape.dims(); ++dim) { in MakeIndexSuffixes()
196 for (int i = 0; i < values_shape.dim_size(dim); ++i) { in MakeIndexSuffixes()
Dragged_gather_op.cc240 TensorShape values_shape = params_dense_values_in.shape(); in WriteValues() local
241 values_shape.set_dim(0, num_values); in WriteValues()
243 context->allocate_output(values_index, values_shape, &values_out)); in WriteValues()
Dexample_parsing_ops.cc1052 TensorShape values_shape({0}); in Compute() local
1059 ctx, context_sparse_values.allocate(d, values_shape, &sp_values_d)); in Compute()
1162 TensorShape values_shape({total_num_features}); in Compute() local
1168 OP_REQUIRES_OK(ctx, feature_list_sparse_values.allocate(d, values_shape, in Compute()
/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.cc1264 TensorShape values_shape; in FastParseExample() local
1265 values_shape.AddDim(total_num_features); in FastParseExample()
1266 result->sparse_values.emplace_back(config.sparse[d].dtype, values_shape); in FastParseExample()
1322 TensorShape values_shape; in FastParseExample() local
1323 values_shape.AddDim(total_num_features); in FastParseExample()
1324 result->ragged_values.emplace_back(config.ragged[d].dtype, values_shape); in FastParseExample()
1375 TensorShape values_shape; in FastParseExample() local
1378 values_shape.AddDim(batch_size); in FastParseExample()
1379 values_shape.AddDim(max_num_elements); in FastParseExample()
1381 values_shape.AddDim(config.dense[d].shape.dim_size(i)); in FastParseExample()
[all …]
Dexample_proto_helper.cc315 output_shapes->values_shape.AddDim(total_num_features); in GetSparseTensorShapes()
382 const TensorShape& values_shape = sparse_tensor_batch_shapes.values_shape; in BatchExampleProtoToTensors() local
387 (*output_sparse_values_tensor)[d] = Tensor(allocator, dtype, values_shape); in BatchExampleProtoToTensors()
Dexample_proto_helper.h88 TensorShape values_shape; member
/external/tensorflow/tensorflow/lite/kernels/
Dsparse_to_dense_test.cc34 std::initializer_list<int> values_shape, T default_value, in SparseToDenseOpModel() argument
46 BuildInterpreter({indices_shape, output_shape_shape, values_shape, {1}}); in SparseToDenseOpModel()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_tensor_to_tensor_tflite_test.cc42 std::initializer_list<int> values_shape, in RaggedTensorToTensorOpModel() argument
53 shapes.emplace_back(values_shape); in RaggedTensorToTensorOpModel()
Dragged_tensor_to_tensor_tflite.cc254 RuntimeShape values_shape(values.dims->size, values.dims->data); in CalculateOutputSize() local
258 if (!ValidateDefaultValueShape(context, default_value_shape, values_shape)) { in CalculateOutputSize()
264 ragged_rank, output_shape_shape, values_shape); in CalculateOutputSize()
/external/tensorflow/tensorflow/python/framework/
Dsparse_tensor.py145 values_shape = values.shape.with_rank(1)
149 indices_shape.dims[0].assert_is_compatible_with(values_shape.dims[0])
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.h252 ShapeHandle values_shape, ShapeHandle shape_shape);
Dcommon_shape_fns.cc2228 ShapeHandle values_shape, ShapeHandle shape_shape) { in ValidateSparseTensor() argument
2232 TF_RETURN_IF_ERROR(c->WithRank(values_shape, 1, &unused_shape)); in ValidateSparseTensor()
2238 DimensionHandle num_values_elements_dim = c->Dim(values_shape, 0); in ValidateSparseTensor()
/external/tensorflow/tensorflow/lite/kernels/parse_example/
Dparse_example.cc553 tf::TensorShape values_shape; in FastParseExampleLite() local
556 values_shape.AddDim(batch_size); in FastParseExampleLite()
557 values_shape.AddDim(max_num_elements); in FastParseExampleLite()
559 values_shape.AddDim(config.dense[d].shape.dim_size(i)); in FastParseExampleLite()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc1004 Shape values_shape = ShapeUtil::MakeShape(F32, {2, 3}); in TEST_F() local
1008 auto values = Parameter(&builder, 2, values_shape, "values"); in TEST_F()
Dhlo_dataflow_analysis_test.cc2586 Shape values_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2590 HloInstruction::CreateParameter(1, values_shape, "values")); in TEST_F()
2593 MakeSortHlo(ShapeUtil::MakeTupleShape({keys_shape, values_shape}), in TEST_F()
Dalgebraic_simplifier_test.cc3581 Shape values_shape = ShapeUtil::MakeShape(S32, {5, 0}); in TEST_F() local
3585 HloInstruction::CreateParameter(1, values_shape, "values0")); in TEST_F()
3587 HloInstruction::CreateParameter(2, values_shape, "values1")); in TEST_F()
3589 {keys_shape, values_shape, values_shape}), in TEST_F()
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor.py1685 values_shape = array_ops.concat([[input_shape[0] * input_shape[1]],
1687 values = array_ops.reshape(tensor, values_shape)