Searched defs:SizeOp (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ | ||
H A D | shape_ops.cc | 259 SizeOp<int32>); member in namespace:tensorflow 264 SizeOp<int64>); member in namespace:tensorflow 273 SizeOp<int32>); \ 279 SizeOp<int64>); 294 SizeOp<int32>); member in namespace:tensorflow 301 SizeOp<int64>); member in namespace:tensorflow 312 SizeOp<int32>); \ 318 SizeOp<int64>); 329 SizeOp<int32>); member in namespace:tensorflow 336 SizeOp<int6 member in namespace:tensorflow [all...] |
H A D | shape_ops.h | 125 class SizeOp : public OpKernel { class in namespace:tensorflow 127 explicit SizeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} function in class:tensorflow::SizeOp |
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ | ||
H A D | shape_op.cc | 86 class SizeOp : public XlaOpKernel { class in namespace:tensorflow::__anon25455 88 explicit SizeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} function in class:tensorflow::__anon25455::SizeOp 103 REGISTER_XLA_OP(Name("Size"), SizeOp); member in namespace:tensorflow::__anon25455 |
Completed in 267 milliseconds