Home
last modified time | relevance | path

Searched refs:SetDimWithStatus (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/image/
Dattention_ops.cc92 OP_REQUIRES_OK(context, output_shape.SetDimWithStatus(1, output_height)); in Compute()
93 OP_REQUIRES_OK(context, output_shape.SetDimWithStatus(2, output_width)); in Compute()
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h597 TF_RETURN_IF_ERROR(output_shape.SetDimWithStatus(dim, slice_size)); in Slice()
601 output_shape.SetDimWithStatus(dim, input_size - start_index)); in Slice()
604 TF_RETURN_IF_ERROR(output_shape.SetDimWithStatus(dim, 0)); in Slice()
/external/tensorflow/tensorflow/core/kernels/
Dsegment_reduction_ops_impl.h117 OP_REQUIRES_OK(context, output_shape.SetDimWithStatus(0, output_rows)); in Compute()
295 output_shape.SetDimWithStatus(0, output_rows), done); in ComputeAsync()
583 OP_REQUIRES_OK(context, output_shape.SetDimWithStatus(0, output_rows));
1165 OP_REQUIRES_OK(context, output_shape.SetDimWithStatus(0, M));
/external/tensorflow/tensorflow/core/framework/
Dtensor_shape.h236 Status SetDimWithStatus(int d, int64_t size);
Dtensor_shape.cc596 Status TensorShapeBase<Shape>::SetDimWithStatus(int d, int64_t size) { in SetDimWithStatus() function in tensorflow::TensorShapeBase
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_lib.cc587 combined_shape.SetDimWithStatus(axis_index, axis_length)); in Shape()