Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/
Dweights_broadcast_ops.py30 def _has_valid_dims(weights_shape, values_shape): argument
32 None, "has_invalid_dims", (weights_shape, values_shape)) as scope:
33 values_shape_2d = array_ops.expand_dims(values_shape, -1)
44 weights_rank, weights_shape, values_rank, values_shape): argument
47 (weights_rank, weights_shape, values_rank, values_shape)) as scope:
52 lambda: _has_valid_dims(weights_shape, values_shape),
87 values_shape = array_ops.shape(values, name="shape")
102 values_shape_static = tensor_util.constant_value(values_shape)
122 "values.shape=", values.name, values_shape,
129 weights_rank, weights_shape, values_rank, values_shape),
[all …]
Dwhile_v2_indexed_slices_rewriter.py184 values_shape = tensor_shape.TensorShape([0] +
187 values_shape, dtype=values_out.dtype, name="values_init")
193 values_shape = array_ops.concat([[0], forward_shape[1:]], 0)
195 values_shape, dtype=values_out.dtype, name="values_init")
Darray_grad.py588 values_shape = array_ops.concat([size, params_shape[1:]], 0)
590 _IndexedSlicesToTensorNoWarning(grad), values_shape)
624 values_shape = array_ops.shape(values)
626 outer_shape = values_shape[:batch_dims]
627 inner_shape = values_shape[batch_dims:][1:]
684 values_shape = array_ops.concat([indices_size, params_tail_shape], 0)
686 _IndexedSlicesToTensorNoWarning(grad), values_shape)
695 values_shape = array_ops.concat([outer_shape, [-1], inner_shape], 0)
697 values_dims = array_ops.size(values_shape)
705 _IndexedSlicesToTensorNoWarning(grad), values_shape)
Dcontrol_flow_ops.py2017 values_shape = tensor_shape.TensorShape([tensor_shape.Dimension(1)] +
2022 0, values.dtype, shape=values_shape, name="b_acc")
2026 values_shape = _resource_safe_shape(op.inputs[0])[1:]
2027 values_shape = array_ops.concat([[1], values_shape], 0)
2028 values_acc = array_ops.zeros(values_shape, dtype=values.dtype)
Dresource_variable_ops.py2277 values_shape = array_ops.concat([size, params_shape[1:]], 0)
2278 values = array_ops.reshape(grad, values_shape)
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_variant_op.cc41 TensorShape values_shape = batched_values.shape(); in UnbatchDenseZerothDim() local
42 if (values_shape.dims() < 1) { in UnbatchDenseZerothDim()
45 auto num_components = values_shape.dim_size(0); in UnbatchDenseZerothDim()
46 values_shape.RemoveDim(0); in UnbatchDenseZerothDim()
47 auto num_values = values_shape.num_elements(); in UnbatchDenseZerothDim()
54 Tensor(DataTypeToEnum<VALUE_TYPE>::value, values_shape)); in UnbatchDenseZerothDim()
92 TensorShape values_shape = batched_ragged.values().shape(); in UnbatchRaggedZerothDim() local
100 values_shape.set_dim(0, num_values); in UnbatchRaggedZerothDim()
102 Tensor(DataTypeToEnum<VALUE_TYPE>::value, values_shape)); in UnbatchRaggedZerothDim()
157 values_shape.set_dim(0, num_values); in UnbatchRaggedZerothDim()
[all …]
Dragged_tensor_to_sparse_kernel.cc202 const TensorShape& values_shape) { in MakeIndexSuffixes() argument
204 for (int dim = 1; dim < values_shape.dims(); ++dim) { in MakeIndexSuffixes()
207 for (int i = 0; i < values_shape.dim_size(dim); ++i) { in MakeIndexSuffixes()
Dragged_gather_op.cc246 TensorShape values_shape = params_dense_values_in.shape(); in WriteValues() local
247 values_shape.set_dim(0, num_values); in WriteValues()
249 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/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/util/
Dexample_proto_fast_parsing.cc1280 TensorShape values_shape; in FastParseExample() local
1281 values_shape.AddDim(total_num_features); in FastParseExample()
1282 result->sparse_values.emplace_back(config.sparse[d].dtype, values_shape); in FastParseExample()
1338 TensorShape values_shape; in FastParseExample() local
1339 values_shape.AddDim(total_num_features); in FastParseExample()
1340 result->ragged_values.emplace_back(config.ragged[d].dtype, values_shape); in FastParseExample()
1391 TensorShape values_shape; in FastParseExample() local
1394 values_shape.AddDim(batch_size); in FastParseExample()
1395 values_shape.AddDim(max_num_elements); in FastParseExample()
1397 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/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/lite/kernels/
Dsparse_to_dense_test.cc40 std::initializer_list<int> values_shape, T default_value, in SparseToDenseOpModel() argument
58 BuildInterpreter({indices_shape, output_shape_shape, values_shape, {1}}, in SparseToDenseOpModel()
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc665 auto values_shape = Shape(scope, values); in BatchGatherGrad() local
667 outer_shape = Slice(scope, values_shape, {0}, {batch_dims}); in BatchGatherGrad()
669 Slice(scope, Slice(scope, values_shape, {batch_dims}, {-1}), {1}, {-1}); in BatchGatherGrad()
730 auto values_shape = Concat(scope, {outer_shape, {-1}, inner_shape}, 0); in GatherV2Grad() local
731 auto values_dims = Size(scope, values_shape); in GatherV2Grad()
740 auto values = Reshape(scope, grad_inputs[0], values_shape); in GatherV2Grad()
/external/tensorflow/tensorflow/python/framework/
Dsparse_tensor.py141 values_shape = values.shape.with_rank(1)
145 indices_shape.dims[0].assert_is_compatible_with(values_shape.dims[0])
/external/tensorflow/tensorflow/lite/kernels/parse_example/
Dparse_example.cc555 tf::TensorShape values_shape; in FastParseExampleLite() local
558 values_shape.AddDim(batch_size); in FastParseExampleLite()
559 values_shape.AddDim(max_num_elements); in FastParseExampleLite()
561 values_shape.AddDim(config.dense[d].shape.dim_size(i)); in FastParseExampleLite()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.h258 ShapeHandle values_shape, ShapeHandle shape_shape);
Dcommon_shape_fns.cc2319 ShapeHandle values_shape, ShapeHandle shape_shape) { in ValidateSparseTensor() argument
2323 TF_RETURN_IF_ERROR(c->WithRank(values_shape, 1, &unused_shape)); in ValidateSparseTensor()
2329 DimensionHandle num_values_elements_dim = c->Dim(values_shape, 0); in ValidateSparseTensor()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc1175 Shape values_shape = ShapeUtil::MakeShape(F32, {2, 3}); in TEST_F() local
1179 auto values = Parameter(&builder, 2, values_shape, "values"); in TEST_F()
Dhlo_dataflow_analysis_test.cc2647 Shape values_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2651 HloInstruction::CreateParameter(1, values_shape, "values")); in TEST_F()
2654 MakeSortHlo(ShapeUtil::MakeTupleShape({keys_shape, values_shape}), in TEST_F()
Dalgebraic_simplifier_test.cc4017 Shape values_shape = ShapeUtil::MakeShape(S32, {5, 0}); in TEST_F() local
4021 HloInstruction::CreateParameter(1, values_shape, "values0")); in TEST_F()
4023 HloInstruction::CreateParameter(2, values_shape, "values1")); in TEST_F()
4025 {keys_shape, values_shape, values_shape}), in TEST_F()
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor.py1773 values_shape = array_ops.concat(
1775 values = array_ops.reshape(tensor, values_shape)