strided_slice_op.cc | 41 OP_REQUIRES_OK(ctx, ctx->GetAttr("Index", &index_type_)); 59 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor)); 61 LiteralToHostTensor(end_literal, index_type_, &end_tensor)); 62 OP_REQUIRES_OK(ctx, LiteralToHostTensor(strides_literal, index_type_, 106 DataType index_type_; member in class:tensorflow::__anon38785::StridedSliceOp 123 OP_REQUIRES_OK(ctx, ctx->GetAttr("Index", &index_type_)); 142 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor)); 144 LiteralToHostTensor(end_literal, index_type_, &end_tensor)); 145 OP_REQUIRES_OK(ctx, LiteralToHostTensor(strides_literal, index_type_, 215 DataType index_type_; member in class:tensorflow::__anon38785::StridedSliceGradOp 327 DataType index_type_; member in class:tensorflow::__anon38785::StridedSliceAssignOp [all...] |