Searched refs:ConstantInputAsShape (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | random_ops.cc | 46 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
225 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
264 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
292 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
325 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
|
| D | stateless_random_ops.cc | 153 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
198 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
249 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
290 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
341 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
|
| D | conv_ops.cc | 107 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &input_tensor_shape)); in Compile()
173 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(1, &filter_tensor_shape)); in Compile()
|
| D | stateless_random_ops_v2.cc | 206 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
259 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
320 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
366 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
420 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &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 | 90 OP_REQUIRES_OK(context, context->ConstantInputAsShape(2, &buffer_shape)); in Compile()
|
| D | stateful_random_ops.cc | 195 TF_RETURN_IF_ERROR(ctx->ConstantInputAsShape(shape_input_idx, &shape)); in CompileImpl()
|
| D | pooling_ops.cc | 426 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &gradients_shape)); in Compile()
|
| D | strided_slice_op.cc | 485 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &input_shape)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 154 Status ConstantInputAsShape(int index, TensorShape* shape);
|
| D | xla_op_kernel.cc | 405 Status XlaOpKernelContext::ConstantInputAsShape(int index, TensorShape* shape) { in ConstantInputAsShape() function in tensorflow::XlaOpKernelContext
|