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

  /external/tensorflow/tensorflow/core/kernels/
shape_ops.cc 259 SizeOp<int32>);
264 SizeOp<int64>);
273 SizeOp<int32>); \
279 SizeOp<int64>);
294 SizeOp<int32>);
301 SizeOp<int64>);
312 SizeOp<int32>); \
318 SizeOp<int64>);
329 SizeOp<int32>);
336 SizeOp<int64>)
    [all...]
shape_ops.h 125 class SizeOp : public OpKernel {
127 explicit SizeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {}
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
shape_op.cc 86 class SizeOp : public XlaOpKernel {
88 explicit SizeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {}
103 REGISTER_XLA_OP(Name("Size"), SizeOp);

Completed in 172 milliseconds