| /external/tensorflow/tensorflow/core/util/ |
| D | image_resizer_state.h | 149 OP_REQUIRES_OK(context, shape.AddDimWithStatus(batch_size)); in ValidateAndCreateOutput()
150 OP_REQUIRES_OK(context, shape.AddDimWithStatus(out_height)); in ValidateAndCreateOutput()
151 OP_REQUIRES_OK(context, shape.AddDimWithStatus(out_width)); in ValidateAndCreateOutput()
152 OP_REQUIRES_OK(context, shape.AddDimWithStatus(channels)); in ValidateAndCreateOutput()
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2021-071.md | 62 status.Update(AddDimWithStatus(internal::SubtleMustCopy(s)));
70 Status TensorShapeBase<Shape>::AddDimWithStatus(int64 size) {
88 use `BuildTensorShapeBase` or `AddDimWithStatus` to prevent `CHECK`-failures in
|
| D | tfsa-2021-012.md | 57 status.Update(AddDimWithStatus(internal::SubtleMustCopy(s)));
65 Status TensorShapeBase<Shape>::AddDimWithStatus(int64 size) {
83 use `BuildTensorShapeBase` or `AddDimWithStatus` to prevent `CHECK`-failures in
|
| D | tfsa-2021-024.md | 61 status.Update(AddDimWithStatus(internal::SubtleMustCopy(s)));
69 Status TensorShapeBase<Shape>::AddDimWithStatus(int64 size) {
87 use `BuildTensorShapeBase` or `AddDimWithStatus` to prevent `CHECK`-failures in
|
| D | tfsa-2021-200.md | 19 …` failure which provokes a `std::abort`. Instead, code should use [`AddDimWithStatus`](https://git…
|
| D | tfsa-2021-198.md | 9 …/framework/tensor_shape.cc#L395-L408) calls should be replaced by [`AddDimWithStatus`](https://git…
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape.cc | 173 s = out->AddDimWithStatus(d.size()); in BuildTensorShapeBase()
288 status.Update(AddDimWithStatus(internal::SubtleMustCopy(s))); in InitDims()
410 Status TensorShapeBase<Shape>::AddDimWithStatus(int64_t size) { in AddDimWithStatus() function in tensorflow::TensorShapeBase
498 s.Update(AddDimWithStatus(d.size)); in AppendShapeWithStatus()
550 s.Update(AddDimWithStatus(dval)); in InsertDimWithStatus()
625 s.Update(AddDimWithStatus(dval)); in SetDimWithStatus()
690 s.Update(AddDimWithStatus(dval)); in RemoveDimRangeWithStatus()
870 return out->AddDimWithStatus(size); in ConcatenateWithStatus()
891 Status s = out->AddDimWithStatus(dim.size); in ConcatenateWithStatus()
930 s.Update(result->AddDimWithStatus(dim0 >= 0 ? dim0 : dim1)); in MergeWith()
|
| D | tensor_shape_test.cc | 209 TEST(TensorShapeTest, AddDimWithStatus) { in TEST() argument
211 Status status = s.AddDimWithStatus(400); in TEST()
216 status = s.AddDimWithStatus(-1); in TEST()
|
| D | tensor_shape.h | 209 Status AddDimWithStatus(int64_t size);
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | crop_and_resize_op.cc | 174 OP_REQUIRES_OK_ASYNC(context, shape.AddDimWithStatus(num_boxes), done); in ComputeAsync()
175 OP_REQUIRES_OK_ASYNC(context, shape.AddDimWithStatus(crop_height), done); in ComputeAsync()
176 OP_REQUIRES_OK_ASYNC(context, shape.AddDimWithStatus(crop_width), done); in ComputeAsync()
177 OP_REQUIRES_OK_ASYNC(context, shape.AddDimWithStatus(depth), done); in ComputeAsync()
422 OP_REQUIRES_OK_ASYNC(context, shape.AddDimWithStatus(batch_size), done); in ComputeAsync()
423 OP_REQUIRES_OK_ASYNC(context, shape.AddDimWithStatus(image_height), done); in ComputeAsync()
424 OP_REQUIRES_OK_ASYNC(context, shape.AddDimWithStatus(image_width), done); in ComputeAsync()
425 OP_REQUIRES_OK_ASYNC(context, shape.AddDimWithStatus(depth), done); in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op.cc | 307 OP_REQUIRES_OK(context, output_shape.AddDimWithStatus(shape_vec(i))); in Compute()
466 OP_REQUIRES_OK(context, output_shape.AddDimWithStatus(shape_vec(i))); in Compute()
552 OP_REQUIRES_OK(context, output_shape.AddDimWithStatus(shape_vec(i))); in Compute()
|
| D | sparse_split_op.cc | 126 context, dense_shape.AddDimWithStatus(input_shape_flat(i)), done); in SparseSplitOpImpl()
|
| D | sequence_ops.cc | 110 OP_REQUIRES_OK(context, shape.AddDimWithStatus(size)); in Compute()
|
| D | pad_op.cc | 89 context, output_shape.AddDimWithStatus(before_d + size_d + after_d)); in Compute()
|
| D | depthwise_conv_grad_op.cc | 628 OP_REQUIRES_OK(context, input_shape.AddDimWithStatus(in_sizes_data[i])); in Compute()
1133 filter_shape.AddDimWithStatus(filter_sizes_data[i])); in Compute()
|
| D | tile_ops.cc | 191 OP_REQUIRES_OK(context, output_shape.AddDimWithStatus( in Compute()
|
| D | pooling_ops_3d.cc | 526 OP_REQUIRES_OK(context, output_shape.AddDimWithStatus(shape_vec(i))); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_avgpooling_op.cc | 207 orig_input_shape.AddDimWithStatus(shape_vec(i)); in Compute()
|