Home
last modified time | relevance | path

Searched defs:EnsureShapeOp (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Densure_shape_op.cc33 class EnsureShapeOp : public XlaOpKernel { class
35 explicit EnsureShapeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in EnsureShapeOp() function in tensorflow::__anonc6983b9b0111::EnsureShapeOp
/external/tensorflow/tensorflow/core/kernels/
Dshape_ops.cc471 class EnsureShapeOp : public OpKernel { class
473 explicit EnsureShapeOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in EnsureShapeOp() function in tensorflow::EnsureShapeOp