Searched refs:ConstantInputAsShape (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | stateless_random_ops.cc | 79 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
118 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
166 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
207 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
|
| D | random_ops.cc | 44 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
214 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
249 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
277 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
|
| D | conv_ops.cc | 102 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &input_tensor_shape)); in Compile()
160 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(1, &filter_tensor_shape)); in Compile()
|
| D | broadcast_to_op.cc | 33 OP_REQUIRES_OK(context, context->ConstantInputAsShape(1, &output_shape)); in Compile()
|
| D | sparse_to_dense_op.cc | 44 OP_REQUIRES_OK(context, context->ConstantInputAsShape(1, &output_shape)); in Compile()
|
| D | scatter_nd_op.cc | 88 OP_REQUIRES_OK(context, context->ConstantInputAsShape(2, &buffer_shape)); in Compile()
|
| D | strided_slice_op.cc | 135 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &input_shape)); in Compile()
|
| D | stateful_random_ops.cc | 214 TF_RETURN_IF_ERROR(ctx->ConstantInputAsShape(shape_input_idx, &shape)); in CompileImpl()
|
| D | pooling_ops.cc | 405 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &gradients_shape)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 139 Status ConstantInputAsShape(int index, TensorShape* shape);
|
| D | xla_op_kernel.cc | 313 Status XlaOpKernelContext::ConstantInputAsShape(int index, TensorShape* shape) { in ConstantInputAsShape() function in tensorflow::XlaOpKernelContext
|