Searched refs:BuildTensorShapeBase (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | tensor_shape.h | 183 static Status BuildTensorShapeBase(gtl::ArraySlice<int64_t> dim_sizes,
185 static Status BuildTensorShapeBase(std::initializer_list<int64_t> dim_sizes, in BuildTensorShapeBase() function
187 return BuildTensorShapeBase(gtl::ArraySlice<int64_t>(dim_sizes), out); in BuildTensorShapeBase()
189 static Status BuildTensorShapeBase(const TensorShapeProto& proto,
368 return BuildTensorShapeBase(dim_sizes, out); in BuildTensorShape()
376 return BuildTensorShapeBase(proto, out); in BuildTensorShape()
543 return BuildTensorShapeBase(dim_sizes, out); in BuildPartialTensorShape()
551 return BuildTensorShapeBase(proto, out); in BuildPartialTensorShape()
557 TF_RETURN_IF_ERROR(BuildTensorShapeBase(proto, &out)); in BuildPartialTensorShape()
|
| H A D | tensor_shape_test.cc | 222 Status status = TensorShape::BuildTensorShapeBase({10, 5, 20}, &s); in TEST()
227 status = TensorShape::BuildTensorShapeBase({-10, 5, 20}, &s); in TEST()
|
| H A D | tensor_shape.cc | 158 Status TensorShapeBase<Shape>::BuildTensorShapeBase( in BuildTensorShapeBase() function in tensorflow::TensorShapeBase
190 Status TensorShapeBase<Shape>::BuildTensorShapeBase( in BuildTensorShapeBase() function in tensorflow::TensorShapeBase
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_slice_op.cc | 45 TensorShapeBase<TensorShape>::BuildTensorShapeBase( in operator ()()
65 OP_REQUIRES_OK(context, TensorShapeBase<TensorShape>::BuildTensorShapeBase( in operator ()()
69 OP_REQUIRES_OK(context, TensorShapeBase<TensorShape>::BuildTensorShapeBase( in operator ()()
|
| /aosp_15_r20/external/tensorflow/tensorflow/security/advisory/ |
| H A D | tfsa-2021-012.md | 83 use `BuildTensorShapeBase` or `AddDimWithStatus` to prevent `CHECK`-failures in
|
| H A D | tfsa-2021-071.md | 88 use `BuildTensorShapeBase` or `AddDimWithStatus` to prevent `CHECK`-failures in
|
| H A D | tfsa-2021-024.md | 87 use `BuildTensorShapeBase` or `AddDimWithStatus` to prevent `CHECK`-failures in
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | tensor_slice_reader.cc | 173 status_ = TensorShape::BuildTensorShapeBase(ssm.shape(), &ssm_shape); in LoadShard()
|