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

  /external/tensorflow/tensorflow/core/kernels/
reshape_op.cc 25 ReshapeOp);
30 ReshapeOp);
38 ReshapeOp); \
44 ReshapeOp);
56 ReshapeOp); \
62 ReshapeOp);
77 ReshapeOp);
85 ReshapeOp);
100 ReshapeOp);
108 ReshapeOp);
    [all...]
reshape_op.h 31 class ReshapeOp : public OpKernel {
33 explicit ReshapeOp(OpKernelConstruction* context) : OpKernel(context) {}
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
reshape_op.cc 30 class ReshapeOp : public XlaOpKernel {
32 explicit ReshapeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {}
98 REGISTER_XLA_OP(Name("Reshape").CompileTimeConstInput("shape"), ReshapeOp);

Completed in 57 milliseconds