Home
last modified time | relevance | path

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

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