Searched refs:ExpandDimsOp (Results 1 – 3 of 3) sorted by relevance
345 ExpandDimsOp<int32>);350 ExpandDimsOp<int64>);359 ExpandDimsOp<int32>); \365 ExpandDimsOp<int64>);377 ExpandDimsOp<int32>);385 ExpandDimsOp<int64>);395 ExpandDimsOp<int32>); \401 ExpandDimsOp<int64>);413 ExpandDimsOp<int32>);421 ExpandDimsOp<int64>);
136 class ExpandDimsOp : public OpKernel {138 explicit ExpandDimsOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in ExpandDimsOp() function
114 class ExpandDimsOp : public XlaOpKernel { class116 explicit ExpandDimsOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ExpandDimsOp() function in tensorflow::__anon812f91ee0111::ExpandDimsOp158 ExpandDimsOp);