HomeSort by relevance Sort by last modified time
    Searched refs:ShapeNOp (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
shape_ops.cc 115 ShapeNOp<int32>);
120 ShapeNOp<int64>);
129 ShapeNOp<int32>); \
135 ShapeNOp<int64>)
150 ShapeNOp<int32>);
157 ShapeNOp<int64>);
167 ShapeNOp<int32>); \
173 ShapeNOp<int64>)
185 ShapeNOp<int32>);
192 ShapeNOp<int64>)
    [all...]
shape_ops.h 77 class ShapeNOp : public OpKernel {
79 explicit ShapeNOp(OpKernelConstruction* ctx) : OpKernel(ctx) {}
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
shape_op.cc 48 class ShapeNOp : public XlaOpKernel {
50 explicit ShapeNOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {
68 REGISTER_XLA_OP(Name("ShapeN"), ShapeNOp);

Completed in 201 milliseconds